This means that they only generate tests if they are called as the main
script, but not if they are imported as modules. This is important for
testing purposes.

Signed-off-by: Dylan Baker <dylanx.c.ba...@intel.com>
---
 generated_tests/gen_cl_int_builtins.py        |  4 +-
 generated_tests/gen_cl_math_builtins.py       |  3 +-
 generated_tests/gen_cl_relational_builtins.py |  3 +-
 generated_tests/gen_cl_store_tests.py         | 79 ++++++++++++++-------------
 4 files changed, 49 insertions(+), 40 deletions(-)

diff --git a/generated_tests/gen_cl_int_builtins.py 
b/generated_tests/gen_cl_int_builtins.py
index 61fc93e..2e10f92 100644
--- a/generated_tests/gen_cl_int_builtins.py
+++ b/generated_tests/gen_cl_int_builtins.py
@@ -560,4 +560,6 @@ def main():
 
     gen(DATA_TYPES, CLC_VERSION_MIN, functions, testDefs, dirName)
 
-main()
+
+if __name__ == '__main__':
+    main()
diff --git a/generated_tests/gen_cl_math_builtins.py 
b/generated_tests/gen_cl_math_builtins.py
index 1c96adb..332cec9 100644
--- a/generated_tests/gen_cl_math_builtins.py
+++ b/generated_tests/gen_cl_math_builtins.py
@@ -402,4 +402,5 @@ def main():
     gen(DATA_TYPES, CLC_VERSION_MIN, functions, testDefs, dirName)
 
 
-main()
+if __name__ == '__main__':
+    main()
diff --git a/generated_tests/gen_cl_relational_builtins.py 
b/generated_tests/gen_cl_relational_builtins.py
index eaa1775..d8d109f 100644
--- a/generated_tests/gen_cl_relational_builtins.py
+++ b/generated_tests/gen_cl_relational_builtins.py
@@ -191,4 +191,5 @@ def main():
     gen(DATA_TYPES, CLC_VERSION_MIN, functions, testDefs, dirName)
 
 
-main()
+if __name__ == '__main__':
+    main()
diff --git a/generated_tests/gen_cl_store_tests.py 
b/generated_tests/gen_cl_store_tests.py
index e1bc06d..4bd6a46 100644
--- a/generated_tests/gen_cl_store_tests.py
+++ b/generated_tests/gen_cl_store_tests.py
@@ -63,40 +63,45 @@ def begin_test(type_name, addr_space):
     return f
 
 
-for t in TYPES:
-    for s in VEC_SIZES:
-        if s == '':
-            size = 1
-        else:
-            size = int(s)
-        type_name = t + s
-        f = begin_test(type_name, 'global')
-        f.write(textwrap.dedent("""
-        [test]
-        name: global address space
-        global_size: 1 0 0
-        kernel_name: store_global
-        arg_out: 0 buffer {type_name}[8] {gen_array}
-        arg_in:  1 buffer {type_name}[8] {gen_array}
-        [test]
-        name: global address space work items
-        global_size: 8 0 0
-        kernel_name: store_global_wi
-        arg_out: 0 buffer {type_name}[8] {gen_array}
-        arg_in:  1 buffer {type_name}[8] {gen_array}
-        """.format(type_name=type_name, gen_array=gen_array(size))))
-
-        f.close()
-
-        f = begin_test(type_name, 'local')
-        f.write(textwrap.dedent("""
-        [test]
-        name: local address space
-        global_size: 8 0 0
-        local_size:  8 0 0
-        kernel_name: store_local
-        arg_out: 0 buffer {type_name}[8] {gen_array}
-        arg_in:  1 buffer {type_name}[8] {gen_array}
-        """.format(type_name=type_name, gen_array=gen_array(size))))
-
-        f.close()
+def main():
+    for t in TYPES:
+        for s in VEC_SIZES:
+            if s == '':
+                size = 1
+            else:
+                size = int(s)
+            type_name = t + s
+            f = begin_test(type_name, 'global')
+            f.write(textwrap.dedent("""
+            [test]
+            name: global address space
+            global_size: 1 0 0
+            kernel_name: store_global
+            arg_out: 0 buffer {type_name}[8] {gen_array}
+            arg_in:  1 buffer {type_name}[8] {gen_array}
+            [test]
+            name: global address space work items
+            global_size: 8 0 0
+            kernel_name: store_global_wi
+            arg_out: 0 buffer {type_name}[8] {gen_array}
+            arg_in:  1 buffer {type_name}[8] {gen_array}
+            """.format(type_name=type_name, gen_array=gen_array(size))))
+
+            f.close()
+
+            f = begin_test(type_name, 'local')
+            f.write(textwrap.dedent("""
+            [test]
+            name: local address space
+            global_size: 8 0 0
+            local_size:  8 0 0
+            kernel_name: store_local
+            arg_out: 0 buffer {type_name}[8] {gen_array}
+            arg_in:  1 buffer {type_name}[8] {gen_array}
+            """.format(type_name=type_name, gen_array=gen_array(size))))
+
+            f.close()
+
+
+if __name__ == '__main__':
+    main()
-- 
2.3.0

_______________________________________________
Piglit mailing list
Piglit@lists.freedesktop.org
http://lists.freedesktop.org/mailman/listinfo/piglit

Reply via email to