Add function

2022-10-31 13:02:05 +01:00 · 2022-10-31 13:02:05 +01:00 · f04e1cda4b
commit f04e1cda4b
parent 60485311c8
1 changed files with 6 additions and 1 deletions
--- a/tests/mgbench/dataset_creator.py
+++ b/tests/mgbench/dataset_creator.py
@ -47,7 +47,8 @@ import argparse
 #
 # AccessControl specific: uuid is the schema

-if __name__ == "__main__":
+
+def main():
    parser = argparse.ArgumentParser()
    parser.add_argument("--number_of_identities", type=int, default=10)
    parser.add_argument("--number_of_files", type=int, default=10)
@ -113,3 +114,7 @@ if __name__ == "__main__":
                uuid += 1

    f.close()
+
+
+if __name__ == "__main__":
+    main()