better __main__ tests