Add if __name__ ... part at the end, see other test files.