def square(x): return x ** 2 def test_square(): x = 4 # a chosen x-value expected = 16 # expected result computed = square(x) success = computed == expected # boolean value: test passed? msg = f'computed {computed}, expected {expected}' assert success, msg test_square()