warp-lang 1.0.0b5__py3-none-manylinux2014_x86_64.whl → 1.0.0b6__py3-none-manylinux2014_x86_64.whl

warp/tests/test_quat.py

@@ -1,9 +1,16 @@
+# Copyright (c) 2022 NVIDIA CORPORATION.  All rights reserved.
+# NVIDIA CORPORATION and its licensors retain all intellectual property
+# and proprietary rights in and to this software, related documentation
+# and any modifications thereto.  Any use, reproduction, disclosure or
+# distribution of this software and related documentation without an express
+# license agreement from NVIDIA CORPORATION is strictly prohibited.
+
 import unittest
 
 import numpy as np
 
 import warp as wp
-from warp.tests.test_base import *
+from warp.tests.unittest_utils import *
 
 wp.init()
 
@@ -1932,99 +1939,125 @@ def test_constructor_default():
     wp.expect_eq(qeye[3], 1.0)
 
 
-def register(parent):
-    devices = get_test_devices()
+def test_py_arithmetic_ops(test, device, dtype):
+    wptype = wp.types.np_dtype_to_warp_type[np.dtype(dtype)]
 
-    class TestQuat(parent):
-        pass
+    def make_quat(*args):
+        if wptype in wp.types.int_types:
+            # Cast to the correct integer type to simulate wrapping.
+            return tuple(wptype._type_(x).value for x in args)
 
-    add_kernel_test(TestQuat, test_constructor_default, dim=1, devices=devices)
+        return args
 
-    for dtype in np_float_types:
-        add_function_test_register_kernel(
-            TestQuat, f"test_constructors_{dtype.__name__}", test_constructors, devices=devices, dtype=dtype
-        )
-        add_function_test_register_kernel(
-            TestQuat,
-            f"test_casting_constructors_{dtype.__name__}",
-            test_casting_constructors,
-            devices=devices,
-            dtype=dtype,
-        )
-        add_function_test_register_kernel(
-            TestQuat, f"test_anon_type_instance_{dtype.__name__}", test_anon_type_instance, devices=devices, dtype=dtype
-        )
-        add_function_test_register_kernel(
-            TestQuat, f"test_inverse_{dtype.__name__}", test_inverse, devices=devices, dtype=dtype
-        )
-        add_function_test_register_kernel(
-            TestQuat, f"test_quat_identity_{dtype.__name__}", test_quat_identity, devices=devices, dtype=dtype
-        )
-        add_function_test_register_kernel(
-            TestQuat, f"test_dotproduct_{dtype.__name__}", test_dotproduct, devices=devices, dtype=dtype
-        )
-        add_function_test_register_kernel(
-            TestQuat, f"test_length_{dtype.__name__}", test_length, devices=devices, dtype=dtype
-        )
-        add_function_test_register_kernel(
-            TestQuat, f"test_normalize_{dtype.__name__}", test_normalize, devices=devices, dtype=dtype
-        )
-        add_function_test_register_kernel(
-            TestQuat, f"test_addition_{dtype.__name__}", test_addition, devices=devices, dtype=dtype
-        )
-        add_function_test_register_kernel(
-            TestQuat, f"test_subtraction_{dtype.__name__}", test_subtraction, devices=devices, dtype=dtype
-        )
-        add_function_test_register_kernel(
-            TestQuat,
-            f"test_scalar_multiplication_{dtype.__name__}",
-            test_scalar_multiplication,
-            devices=devices,
-            dtype=dtype,
-        )
-        add_function_test_register_kernel(
-            TestQuat, f"test_scalar_division_{dtype.__name__}", test_scalar_division, devices=devices, dtype=dtype
-        )
-        add_function_test_register_kernel(
-            TestQuat,
-            f"test_quat_multiplication_{dtype.__name__}",
-            test_quat_multiplication,
-            devices=devices,
-            dtype=dtype,
-        )
-        add_function_test_register_kernel(
-            TestQuat, f"test_indexing_{dtype.__name__}", test_indexing, devices=devices, dtype=dtype
-        )
-        add_function_test_register_kernel(
-            TestQuat, f"test_quat_lerp_{dtype.__name__}", test_quat_lerp, devices=devices, dtype=dtype
-        )
-        add_function_test_register_kernel(
-            TestQuat,
-            f"test_quat_to_axis_angle_grad_{dtype.__name__}",
-            test_quat_to_axis_angle_grad,
-            devices=devices,
-            dtype=dtype,
-        )
-        add_function_test_register_kernel(
-            TestQuat, f"test_slerp_grad_{dtype.__name__}", test_slerp_grad, devices=devices, dtype=dtype
-        )
-        add_function_test_register_kernel(
-            TestQuat, f"test_quat_rpy_grad_{dtype.__name__}", test_quat_rpy_grad, devices=devices, dtype=dtype
-        )
-        add_function_test_register_kernel(
-            TestQuat, f"test_quat_from_matrix_{dtype.__name__}", test_quat_from_matrix, devices=devices, dtype=dtype
-        )
-        add_function_test_register_kernel(
-            TestQuat, f"test_quat_rotate_{dtype.__name__}", test_quat_rotate, devices=devices, dtype=dtype
-        )
-        add_function_test_register_kernel(
-            TestQuat, f"test_quat_to_matrix_{dtype.__name__}", test_quat_to_matrix, devices=devices, dtype=dtype
-        )
+    quat_cls = wp.types.quaternion(wptype)
+
+    v = quat_cls(1, -2, 3, -4)
+    test.assertSequenceEqual(+v, make_quat(1, -2, 3, -4))
+    test.assertSequenceEqual(-v, make_quat(-1, 2, -3, 4))
+    test.assertSequenceEqual(v + quat_cls(5, 5, 5, 5), make_quat(6, 3, 8, 1))
+    test.assertSequenceEqual(v - quat_cls(5, 5, 5, 5), make_quat(-4, -7, -2, -9))
+
+    v = quat_cls(2, 4, 6, 8)
+    test.assertSequenceEqual(v * wptype(2), make_quat(4, 8, 12, 16))
+    test.assertSequenceEqual(wptype(2) * v, make_quat(4, 8, 12, 16))
+    test.assertSequenceEqual(v / wptype(2), make_quat(1, 2, 3, 4))
+    test.assertSequenceEqual(wptype(24) / v, make_quat(12, 6, 4, 3))
+
+
+devices = get_test_devices()
 
-    return TestQuat
+
+class TestQuat(unittest.TestCase):
+    pass
+
+
+add_kernel_test(TestQuat, test_constructor_default, dim=1, devices=devices)
+
+for dtype in np_float_types:
+    add_function_test_register_kernel(
+        TestQuat, f"test_constructors_{dtype.__name__}", test_constructors, devices=devices, dtype=dtype
+    )
+    add_function_test_register_kernel(
+        TestQuat,
+        f"test_casting_constructors_{dtype.__name__}",
+        test_casting_constructors,
+        devices=devices,
+        dtype=dtype,
+    )
+    add_function_test_register_kernel(
+        TestQuat, f"test_anon_type_instance_{dtype.__name__}", test_anon_type_instance, devices=devices, dtype=dtype
+    )
+    add_function_test_register_kernel(
+        TestQuat, f"test_inverse_{dtype.__name__}", test_inverse, devices=devices, dtype=dtype
+    )
+    add_function_test_register_kernel(
+        TestQuat, f"test_quat_identity_{dtype.__name__}", test_quat_identity, devices=devices, dtype=dtype
+    )
+    add_function_test_register_kernel(
+        TestQuat, f"test_dotproduct_{dtype.__name__}", test_dotproduct, devices=devices, dtype=dtype
+    )
+    add_function_test_register_kernel(
+        TestQuat, f"test_length_{dtype.__name__}", test_length, devices=devices, dtype=dtype
+    )
+    add_function_test_register_kernel(
+        TestQuat, f"test_normalize_{dtype.__name__}", test_normalize, devices=devices, dtype=dtype
+    )
+    add_function_test_register_kernel(
+        TestQuat, f"test_addition_{dtype.__name__}", test_addition, devices=devices, dtype=dtype
+    )
+    add_function_test_register_kernel(
+        TestQuat, f"test_subtraction_{dtype.__name__}", test_subtraction, devices=devices, dtype=dtype
+    )
+    add_function_test_register_kernel(
+        TestQuat,
+        f"test_scalar_multiplication_{dtype.__name__}",
+        test_scalar_multiplication,
+        devices=devices,
+        dtype=dtype,
+    )
+    add_function_test_register_kernel(
+        TestQuat, f"test_scalar_division_{dtype.__name__}", test_scalar_division, devices=devices, dtype=dtype
+    )
+    add_function_test_register_kernel(
+        TestQuat,
+        f"test_quat_multiplication_{dtype.__name__}",
+        test_quat_multiplication,
+        devices=devices,
+        dtype=dtype,
+    )
+    add_function_test_register_kernel(
+        TestQuat, f"test_indexing_{dtype.__name__}", test_indexing, devices=devices, dtype=dtype
+    )
+    add_function_test_register_kernel(
+        TestQuat, f"test_quat_lerp_{dtype.__name__}", test_quat_lerp, devices=devices, dtype=dtype
+    )
+    add_function_test_register_kernel(
+        TestQuat,
+        f"test_quat_to_axis_angle_grad_{dtype.__name__}",
+        test_quat_to_axis_angle_grad,
+        devices=devices,
+        dtype=dtype,
+    )
+    add_function_test_register_kernel(
+        TestQuat, f"test_slerp_grad_{dtype.__name__}", test_slerp_grad, devices=devices, dtype=dtype
+    )
+    add_function_test_register_kernel(
+        TestQuat, f"test_quat_rpy_grad_{dtype.__name__}", test_quat_rpy_grad, devices=devices, dtype=dtype
+    )
+    add_function_test_register_kernel(
+        TestQuat, f"test_quat_from_matrix_{dtype.__name__}", test_quat_from_matrix, devices=devices, dtype=dtype
+    )
+    add_function_test_register_kernel(
+        TestQuat, f"test_quat_rotate_{dtype.__name__}", test_quat_rotate, devices=devices, dtype=dtype
+    )
+    add_function_test_register_kernel(
+        TestQuat, f"test_quat_to_matrix_{dtype.__name__}", test_quat_to_matrix, devices=devices, dtype=dtype
+    )
+    add_function_test(
+        TestQuat, f"test_py_arithmetic_ops_{dtype.__name__}", test_py_arithmetic_ops, devices=None, dtype=dtype
+    )
 
 
 if __name__ == "__main__":
     wp.build.clear_kernel_cache()
-    _ = register(unittest.TestCase)
     unittest.main(verbosity=2)

warp/tests/test_rand.py

@@ -9,10 +9,11 @@ import unittest
 
 import numpy as np
 
+import warp as wp
+from warp.tests.unittest_utils import *
+
 # import matplotlib.pyplot as plt
 
-import warp as wp
-from warp.tests.test_base import *
 
 wp.init()
 
@@ -62,7 +63,7 @@ def test_rand(test, device):
     wp.copy(int_ab_host, int_ab_device)
     wp.copy(float_01_host, float_01_device)
     wp.copy(float_ab_host, float_ab_device)
-    wp.synchronize()
+    wp.synchronize_device(device)
 
     int_a = int_a_host.numpy()
     int_ab = int_ab_host.numpy()
@@ -269,21 +270,19 @@ def test_poisson(test, device):
     test.assertTrue(np.abs(poisson_high_std - np_poisson_high_std) <= 2e-1)
 
 
-def register(parent):
-    devices = get_test_devices()
+devices = get_test_devices()
+
 
-    class TestNoise(parent):
-        pass
+class TestRand(unittest.TestCase):
+    pass
 
-    add_function_test(TestNoise, "test_rand", test_rand, devices=devices)
-    add_function_test(TestNoise, "test_sample_cdf", test_sample_cdf, devices=devices)
-    add_function_test(TestNoise, "test_sampling_methods", test_sampling_methods, devices=devices)
-    add_function_test(TestNoise, "test_poisson", test_poisson, devices=devices)
 
-    return TestNoise
+add_function_test(TestRand, "test_rand", test_rand, devices=devices)
+add_function_test(TestRand, "test_sample_cdf", test_sample_cdf, devices=devices)
+add_function_test(TestRand, "test_sampling_methods", test_sampling_methods, devices=devices)
+add_function_test(TestRand, "test_poisson", test_poisson, devices=devices)
 
 
 if __name__ == "__main__":
     wp.build.clear_kernel_cache()
-    _ = register(unittest.TestCase)
     unittest.main(verbosity=2)

warp/tests/test_reload.py

@@ -5,25 +5,22 @@
 # distribution of this software and related documentation without an express
 # license agreement from NVIDIA CORPORATION is strictly prohibited.
 
-import numpy as np
-import warp as wp
+import importlib
+import os
+import unittest
 
-import math
+import numpy as np
 
 import warp as wp
-from warp.tests.test_base import *
 
-import unittest
-import importlib
-import os
+# dummy modules used for testing reload with dependencies
+import warp.tests.aux_test_dependent as test_dependent
+import warp.tests.aux_test_reference as test_reference
+import warp.tests.aux_test_reference_reference as test_reference_reference
 
 # dummy module used for testing reload
-import warp.tests.test_square as test_square
-
-# dummy modules used for testing reload with dependencies
-import warp.tests.test_dependent as test_dependent
-import warp.tests.test_reference as test_reference
-import warp.tests.test_reference_reference as test_reference_reference
+import warp.tests.aux_test_square as test_square
+from warp.tests.unittest_utils import *
 
 wp.init()
 
@@ -110,7 +107,7 @@ def run(expect, device):
 
 def test_reload(test, device):
     # write out the module python and import it
-    f = open(os.path.abspath(os.path.join(os.path.dirname(__file__), "test_square.py")), "w")
+    f = open(os.path.abspath(os.path.join(os.path.dirname(__file__), "aux_test_square.py")), "w")
     f.writelines(square_two)
     f.flush()
     f.close()
@@ -118,7 +115,7 @@ def test_reload(test, device):
     reload_module(test_square)
     test_square.run(expect=4.0, device=device)  # 2*2=4
 
-    f = open(os.path.abspath(os.path.join(os.path.dirname(__file__), "test_square.py")), "w")
+    f = open(os.path.abspath(os.path.join(os.path.dirname(__file__), "aux_test_square.py")), "w")
     f.writelines(square_four)
     f.flush()
     f.close()
@@ -130,12 +127,12 @@ def test_reload(test, device):
 
 def test_reload_class(test, device):
     def test_func():
-        import warp.tests.test_class_kernel
-        from warp.tests.test_class_kernel import ClassKernelTest
-
         import importlib as imp
 
-        imp.reload(warp.tests.test_class_kernel)
+        import warp.tests.aux_test_class_kernel
+        from warp.tests.aux_test_class_kernel import ClassKernelTest
+
+        imp.reload(warp.tests.aux_test_class_kernel)
 
         ctest = ClassKernelTest(device)
         expected = np.zeros((10, 3, 3), dtype=np.float32)
@@ -149,7 +146,7 @@ def test_reload_class(test, device):
 template_ref = """# This file is used to test reloading module references.
 
 import warp as wp
-import warp.tests.test_reference_reference as refref
+import warp.tests.aux_test_reference_reference as refref
 
 wp.init()
 
@@ -173,8 +170,8 @@ def more_magic():
 
 
 def test_reload_references(test, device):
-    path_ref = os.path.abspath(os.path.join(os.path.dirname(__file__), "test_reference.py"))
-    path_refref = os.path.abspath(os.path.join(os.path.dirname(__file__), "test_reference_reference.py"))
+    path_ref = os.path.abspath(os.path.join(os.path.dirname(__file__), "aux_test_reference.py"))
+    path_refref = os.path.abspath(os.path.join(os.path.dirname(__file__), "aux_test_reference_reference.py"))
 
     # rewrite both dependency modules and reload them
     with open(path_ref, "w") as f:
@@ -202,21 +199,19 @@ def test_reload_references(test, device):
     test_dependent.run(expect=4.0, device=device)  # 2 * 2 = 4
 
 
-def register(parent):
-    devices = get_test_devices()
+devices = get_test_devices()
+
 
-    class TestReload(parent):
-        pass
+class TestReload(unittest.TestCase):
+    pass
 
-    add_function_test(TestReload, "test_redefine", test_redefine, devices=devices)
-    add_function_test(TestReload, "test_reload", test_reload, devices=devices)
-    add_function_test(TestReload, "test_reload_class", test_reload_class, devices=devices)
-    add_function_test(TestReload, "test_reload_references", test_reload_references, devices=devices)
 
-    return TestReload
+add_function_test(TestReload, "test_redefine", test_redefine, devices=devices)
+add_function_test(TestReload, "test_reload", test_reload, devices=devices)
+add_function_test(TestReload, "test_reload_class", test_reload_class, devices=devices)
+add_function_test(TestReload, "test_reload_references", test_reload_references, devices=devices)
 
 
 if __name__ == "__main__":
     wp.build.clear_kernel_cache()
-    _ = register(unittest.TestCase)
     unittest.main(verbosity=2, failfast=False)

warp/tests/test_rounding.py

@@ -7,11 +7,11 @@
 
 import unittest
 
-import warp as wp
-from warp.tests.test_base import *
-
 import numpy as np
 
+import warp as wp
+from warp.tests.unittest_utils import *
+
 compare_to_numpy = False
 print_results = False
 
@@ -165,18 +165,15 @@ def test_rounding(test, device):
         print("----------------------------------------------")
 
 
-def register(parent):
-    class TestRounding(parent):
-        pass
+class TestRounding(unittest.TestCase):
+    pass
 
-    devices = get_test_devices()
 
-    add_function_test(TestRounding, "test_rounding", test_rounding, devices=devices)
+devices = get_test_devices()
 
-    return TestRounding
+add_function_test(TestRounding, "test_rounding", test_rounding, devices=devices)
 
 
 if __name__ == "__main__":
     wp.build.clear_kernel_cache()
-    _ = register(unittest.TestCase)
     unittest.main(verbosity=2)