It's fine to make this test a main function. 

if __name__ == "__main__":