def generate_testcase(func):
    def func_wrapper(*args, **kwargs):
        return_value = func(*args, **kwargs)
        print('assert %s(%s) == %s' % (func.__name__, ', '.join(repr(arg) for arg in args), return_value))
        return return_value
    return func_wrapper


@generate_testcase
def square(x):
    return x ** 2

square(3) + square(4) == square(5)


import datetime

@generate_testcase
def should_i_sleep(dt):
    if now.weekday() == 5:
        return False
    if now.hour > 23 and now.weekday() != 4:
        return True
    if now.hour < 6 and now.weekday() != 6:
        return True
    return False
    
now = datetime.datetime.now()
print("Should I sleep now? %s" % ['Nah', 'Probably'][should_i_sleep(now)])

assert square(3) == 9
assert square(4) == 16
assert square(5) == 25
assert should_i_sleep(datetime.datetime(2017, 4, 4, 14, 32, 59, 654172)) == False
Should I sleep now? Nah