diff options
Diffstat (limited to 'examples/select_parser.py')
| -rw-r--r-- | examples/select_parser.py | 8 |
1 files changed, 6 insertions, 2 deletions
diff --git a/examples/select_parser.py b/examples/select_parser.py index 652a448..8779212 100644 --- a/examples/select_parser.py +++ b/examples/select_parser.py @@ -188,7 +188,7 @@ select_stmt << ( select_stmt.ignore(comment) -if __name__ == "__main__": +def main(): tests = """\ select * from xyzzy where z > 100 select * from xyzzy where z > 100 order by zz @@ -233,4 +233,8 @@ if __name__ == "__main__": success, _ = select_stmt.runTests(tests) print("\n{}".format("OK" if success else "FAIL")) - sys.exit(0 if success else 1) + return 0 if success else 1 + + +if __name__ == "__main__": + main() |