diff --git a/test/lvol/lvol_test.py b/test/lvol/lvol_test.py index 497e9c318..3d9b22f95 100755 --- a/test/lvol/lvol_test.py +++ b/test/lvol/lvol_test.py @@ -9,6 +9,7 @@ def check_fail_count(fail_count, num_test): else: print("Test: {num_test} - FAIL".format(num_test=num_test)) + if __name__ == "__main__": rpc_py = None total_size = None diff --git a/test/vhost/common/run_fio.py b/test/vhost/common/run_fio.py index 502409d75..4010df476 100755 --- a/test/vhost/common/run_fio.py +++ b/test/vhost/common/run_fio.py @@ -157,5 +157,6 @@ def main(): print("Running job file: {0}".format(fio_cfg)) run_fio(vms, fio_cfg, out_dir, perf_vmex) + if __name__ == "__main__": sys.exit(main())