Make assume script a little more robust

[alexjurkiewicz]

This script might be sourced from another one which uses set -u. Ensure we don't unconditionally access optional environment variables.

(ignore whitespace to understand the diff better)