#!/bin/bash
source "$(dirname "${BASH_SOURCE[0]}")/common.sh"

out=$(run_py_code "with Measure('Some action...'): print_stderr('okay')" 2>&1)
echo "$out" | grep -qE "^Some action...$"
echo "$out" | grep -qE "^...succeeded in .* sec.$"

out=$(run_py_code "with Measure('Some action...', defer=True): pass" 2>&1)
echo "$out" | grep -qE "Some action... succeeded in .* sec."

out=$(run_py_code "with Measure('Some action...', defer=True): raise UserException('oops')" 2>&1 || true)
echo "$out" | grep -qE "Was running: some action... failed in 0.00 sec."