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

warp/tests/test_fp16.py

@@ -1,9 +1,16 @@
+# Copyright (c) 2023 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()
 
@@ -102,25 +109,22 @@ def test_fp16_grad(test, device):
     assert_np_equal(input.grad.numpy(), np.ones(len(s)) * 2.0)
 
 
-def register(parent):
-    class TestFp16(parent):
-        pass
+class TestFp16(unittest.TestCase):
+    pass
 
-    devices = []
-    if wp.is_cpu_available():
-        devices.append("cpu")
-    for cuda_device in wp.get_cuda_devices():
-        if cuda_device.arch >= 70:
-            devices.append(cuda_device)
 
-    add_function_test(TestFp16, "test_fp16_conversion", test_fp16_conversion, devices=devices)
-    add_function_test(TestFp16, "test_fp16_grad", test_fp16_grad, devices=devices)
-    add_function_test(TestFp16, "test_fp16_kernel_parameter", test_fp16_kernel_parameter, devices=devices)
+devices = []
+if wp.is_cpu_available():
+    devices.append("cpu")
+for cuda_device in get_unique_cuda_test_devices():
+    if cuda_device.arch >= 70:
+        devices.append(cuda_device)
 
-    return TestFp16
+add_function_test(TestFp16, "test_fp16_conversion", test_fp16_conversion, devices=devices)
+add_function_test(TestFp16, "test_fp16_grad", test_fp16_grad, devices=devices)
+add_function_test(TestFp16, "test_fp16_kernel_parameter", test_fp16_kernel_parameter, devices=devices)
 
 
 if __name__ == "__main__":
     wp.build.clear_kernel_cache()
-    _ = register(unittest.TestCase)
     unittest.main(verbosity=2)

warp/tests/test_func.py

@@ -5,14 +5,13 @@
 # distribution of this software and related documentation without an express
 # license agreement from NVIDIA CORPORATION is strictly prohibited.
 
-# include parent path
-import numpy as np
 import math
+import unittest
 
-import warp as wp
-from warp.tests.test_base import *
+import numpy as np
 
-import unittest
+import warp as wp
+from warp.tests.unittest_utils import *
 
 wp.init()
 
@@ -83,169 +82,6 @@ def test_override_func():
     wp.expect_eq(i, 3)
 
 
-def test_native_func_export(test, device):
-    # tests calling native functions from Python
-
-    q = wp.quat(0.0, 0.0, 0.0, 1.0)
-    assert_np_equal(np.array([*q]), np.array([0.0, 0.0, 0.0, 1.0]))
-
-    r = wp.quat_from_axis_angle((1.0, 0.0, 0.0), 2.0)
-    assert_np_equal(np.array([*r]), np.array([0.8414709568023682, 0.0, 0.0, 0.5403022170066833]), tol=1.0e-3)
-
-    q = wp.quat(1.0, 2.0, 3.0, 4.0)
-    q = wp.normalize(q) * 2.0
-    assert_np_equal(
-        np.array([*q]),
-        np.array([0.18257418274879456, 0.3651483654975891, 0.547722578048706, 0.7302967309951782]) * 2.0,
-        tol=1.0e-3,
-    )
-
-    v2 = wp.vec2(1.0, 2.0)
-    v2 = wp.normalize(v2) * 2.0
-    assert_np_equal(np.array([*v2]), np.array([0.4472135901451111, 0.8944271802902222]) * 2.0, tol=1.0e-3)
-
-    v3 = wp.vec3(1.0, 2.0, 3.0)
-    v3 = wp.normalize(v3) * 2.0
-    assert_np_equal(
-        np.array([*v3]), np.array([0.26726123690605164, 0.5345224738121033, 0.8017836809158325]) * 2.0, tol=1.0e-3
-    )
-
-    v4 = wp.vec4(1.0, 2.0, 3.0, 4.0)
-    v4 = wp.normalize(v4) * 2.0
-    assert_np_equal(
-        np.array([*v4]),
-        np.array([0.18257418274879456, 0.3651483654975891, 0.547722578048706, 0.7302967309951782]) * 2.0,
-        tol=1.0e-3,
-    )
-
-    v = wp.vec2(0.0)
-    v += wp.vec2(1.0, 1.0)
-    assert v == wp.vec2(1.0, 1.0)
-    v -= wp.vec2(1.0, 1.0)
-    assert v == wp.vec2(0.0, 0.0)
-    v = wp.vec2(2.0, 2.0) - wp.vec2(1.0, 1.0)
-    assert v == wp.vec2(1.0, 1.0)
-    v *= 2.0
-    assert v == wp.vec2(2.0, 2.0)
-    v = v * 2.0
-    assert v == wp.vec2(4.0, 4.0)
-    v = v / 2.0
-    assert v == wp.vec2(2.0, 2.0)
-    v /= 2.0
-    assert v == wp.vec2(1.0, 1.0)
-    v = -v
-    assert v == wp.vec2(-1.0, -1.0)
-    v = +v
-    assert v == wp.vec2(-1.0, -1.0)
-
-    m22 = wp.mat22(1.0, 2.0, 3.0, 4.0)
-    m22 = m22 + m22
-
-    test.assertEqual(m22[1, 1], 8.0)
-    test.assertEqual(str(m22), "[[2.0, 4.0],\n [6.0, 8.0]]")
-
-    t = wp.transform(
-        wp.vec3(1.0, 2.0, 3.0),
-        wp.quat(4.0, 5.0, 6.0, 7.0),
-    )
-    test.assertSequenceEqual(t, (1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
-    test.assertSequenceEqual(t * (1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0), (396.0, 432.0, 720.0, 56.0, 70.0, 84.0, -28.0))
-    test.assertSequenceEqual(
-        t * wp.transform((1.0, 2.0, 3.0), (4.0, 5.0, 6.0, 7.0)), (396.0, 432.0, 720.0, 56.0, 70.0, 84.0, -28.0)
-    )
-
-    t = wp.transform()
-    test.assertSequenceEqual(t, (0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0))
-
-    t = wp.transform(p=(1.0, 2.0, 3.0), q=(4.0, 5.0, 6.0, 7.0))
-    test.assertSequenceEqual(t, (1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
-
-    t = wp.transform(q=(4.0, 5.0, 6.0, 7.0), p=(1.0, 2.0, 3.0))
-    test.assertSequenceEqual(t, (1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
-
-    t = wp.transform((1.0, 2.0, 3.0), q=(4.0, 5.0, 6.0, 7.0))
-    test.assertSequenceEqual(t, (1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
-
-    t = wp.transform(p=(1.0, 2.0, 3.0))
-    test.assertSequenceEqual(t, (1.0, 2.0, 3.0, 0.0, 0.0, 0.0, 1.0))
-
-    t = wp.transform(q=(4.0, 5.0, 6.0, 7.0))
-    test.assertSequenceEqual(t, (0.0, 0.0, 0.0, 4.0, 5.0, 6.0, 7.0))
-
-    t = wp.transform((1.0, 2.0, 3.0), (4.0, 5.0, 6.0, 7.0))
-    test.assertSequenceEqual(t, (1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
-
-    t = wp.transform(p=wp.vec3(1.0, 2.0, 3.0), q=wp.quat(4.0, 5.0, 6.0, 7.0))
-    test.assertSequenceEqual(t, (1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
-
-    t = wp.transform(1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0)
-    test.assertSequenceEqual(t, (1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
-
-    t = wp.transform(wp.transform(1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
-    test.assertSequenceEqual(t, (1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
-
-    t = wp.transform(*wp.transform(1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
-    test.assertSequenceEqual(t, (1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
-
-    transformf = wp.types.transformation(dtype=float)
-
-    t = wp.transformf((1.0, 2.0, 3.0), (4.0, 5.0, 6.0, 7.0))
-    test.assertSequenceEqual(
-        t + transformf((2.0, 3.0, 4.0), (5.0, 6.0, 7.0, 8.0)),
-        (3.0, 5.0, 7.0, 9.0, 11.0, 13.0, 15.0),
-    )
-    test.assertSequenceEqual(
-        t - transformf((2.0, 3.0, 4.0), (5.0, 6.0, 7.0, 8.0)),
-        (-1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0),
-    )
-
-    f = wp.sin(math.pi * 0.5)
-    test.assertAlmostEqual(f, 1.0, places=3)
-
-    m = wp.mat22(0.0, 0.0, 0.0, 0.0)
-    m += wp.mat22(1.0, 1.0, 1.0, 1.0)
-    assert m == wp.mat22(1.0, 1.0, 1.0, 1.0)
-    m -= wp.mat22(1.0, 1.0, 1.0, 1.0)
-    assert m == wp.mat22(0.0, 0.0, 0.0, 0.0)
-    m = wp.mat22(2.0, 2.0, 2.0, 2.0) - wp.mat22(1.0, 1.0, 1.0, 1.0)
-    assert m == wp.mat22(1.0, 1.0, 1.0, 1.0)
-    m *= 2.0
-    assert m == wp.mat22(2.0, 2.0, 2.0, 2.0)
-    m = m * 2.0
-    assert m == wp.mat22(4.0, 4.0, 4.0, 4.0)
-    m = m / 2.0
-    assert m == wp.mat22(2.0, 2.0, 2.0, 2.0)
-    m /= 2.0
-    assert m == wp.mat22(1.0, 1.0, 1.0, 1.0)
-    m = -m
-    assert m == wp.mat22(-1.0, -1.0, -1.0, -1.0)
-    m = +m
-    assert m == wp.mat22(-1.0, -1.0, -1.0, -1.0)
-    m = m * m
-    assert m == wp.mat22(2.0, 2.0, 2.0, 2.0)
-
-
-def test_native_function_error_resolution(test, device):
-    a = wp.mat22f(1.0, 2.0, 3.0, 4.0)
-    b = wp.mat22d(1.0, 2.0, 3.0, 4.0)
-    with test.assertRaisesRegex(
-        RuntimeError,
-        r"^Couldn't find a function 'mul' compatible with " r"the arguments 'mat22f, mat22d'$",
-    ):
-        a * b
-
-
-def test_user_func_export(test, device):
-    # tests calling overloaded user-defined functions from Python
-    i = custom(1)
-    f = custom(1.0)
-    v = custom(wp.vec3(1.0, 0.0, 0.0))
-
-    test.assertEqual(i, 2)
-    test.assertEqual(f, 2.0)
-    assert_np_equal(np.array([*v]), np.array([2.0, 0.0, 0.0]))
-
-
 def test_func_closure_capture(test, device):
     def make_closure_kernel(func):
         def closure_kernel_fn(data: wp.array(dtype=float), expected: float):
@@ -321,34 +157,181 @@ def test_builtin_shadowing():
     wp.expect_eq(sign(1.23), 123.0)
 
 
-def register(parent):
-    devices = get_test_devices()
-
-    class TestFunc(parent):
-        pass
-
-    add_kernel_test(TestFunc, kernel=test_overload_func, name="test_overload_func", dim=1, devices=devices)
-    add_function_test(TestFunc, func=test_return_func, name="test_return_func", devices=devices)
-    add_kernel_test(TestFunc, kernel=test_override_func, name="test_override_func", dim=1, devices=devices)
-    add_function_test(TestFunc, func=test_native_func_export, name="test_native_func_export", devices=["cpu"])
-    add_function_test(
-        TestFunc,
-        func=test_native_function_error_resolution,
-        name="test_native_function_error_resolution",
-        devices=["cpu"],
-    )
-    add_function_test(TestFunc, func=test_user_func_export, name="test_user_func_export", devices=["cpu"])
-    add_function_test(TestFunc, func=test_func_closure_capture, name="test_func_closure_capture", devices=devices)
-    add_function_test(TestFunc, func=test_multi_valued_func, name="test_multi_valued_func", devices=devices)
-    add_kernel_test(TestFunc, kernel=test_func_defaults, name="test_func_defaults", dim=1, devices=devices)
-    add_kernel_test(TestFunc, kernel=test_builtin_shadowing, name="test_builtin_shadowing", dim=1, devices=devices)
-
-    return TestFunc
+devices = get_test_devices()
+
+
+class TestFunc(unittest.TestCase):
+    def test_user_func_export(self):
+        # tests calling overloaded user-defined functions from Python
+        i = custom(1)
+        f = custom(1.0)
+        v = custom(wp.vec3(1.0, 0.0, 0.0))
+
+        self.assertEqual(i, 2)
+        self.assertEqual(f, 2.0)
+        assert_np_equal(np.array([*v]), np.array([2.0, 0.0, 0.0]))
+
+    def test_native_func_export(self):
+        # tests calling native functions from Python
+
+        q = wp.quat(0.0, 0.0, 0.0, 1.0)
+        assert_np_equal(np.array([*q]), np.array([0.0, 0.0, 0.0, 1.0]))
+
+        r = wp.quat_from_axis_angle(wp.vec3(1.0, 0.0, 0.0), 2.0)
+        assert_np_equal(np.array([*r]), np.array([0.8414709568023682, 0.0, 0.0, 0.5403022170066833]), tol=1.0e-3)
+
+        q = wp.quat(1.0, 2.0, 3.0, 4.0)
+        q = wp.normalize(q) * 2.0
+        assert_np_equal(
+            np.array([*q]),
+            np.array([0.18257418274879456, 0.3651483654975891, 0.547722578048706, 0.7302967309951782]) * 2.0,
+            tol=1.0e-3,
+        )
+
+        v2 = wp.vec2(1.0, 2.0)
+        v2 = wp.normalize(v2) * 2.0
+        assert_np_equal(np.array([*v2]), np.array([0.4472135901451111, 0.8944271802902222]) * 2.0, tol=1.0e-3)
+
+        v3 = wp.vec3(1.0, 2.0, 3.0)
+        v3 = wp.normalize(v3) * 2.0
+        assert_np_equal(
+            np.array([*v3]), np.array([0.26726123690605164, 0.5345224738121033, 0.8017836809158325]) * 2.0, tol=1.0e-3
+        )
+
+        v4 = wp.vec4(1.0, 2.0, 3.0, 4.0)
+        v4 = wp.normalize(v4) * 2.0
+        assert_np_equal(
+            np.array([*v4]),
+            np.array([0.18257418274879456, 0.3651483654975891, 0.547722578048706, 0.7302967309951782]) * 2.0,
+            tol=1.0e-3,
+        )
+
+        v = wp.vec2(0.0)
+        v += wp.vec2(1.0, 1.0)
+        assert v == wp.vec2(1.0, 1.0)
+        v -= wp.vec2(1.0, 1.0)
+        assert v == wp.vec2(0.0, 0.0)
+        v = wp.vec2(2.0, 2.0) - wp.vec2(1.0, 1.0)
+        assert v == wp.vec2(1.0, 1.0)
+        v *= 2.0
+        assert v == wp.vec2(2.0, 2.0)
+        v = v * 2.0
+        assert v == wp.vec2(4.0, 4.0)
+        v = v / 2.0
+        assert v == wp.vec2(2.0, 2.0)
+        v /= 2.0
+        assert v == wp.vec2(1.0, 1.0)
+        v = -v
+        assert v == wp.vec2(-1.0, -1.0)
+        v = +v
+        assert v == wp.vec2(-1.0, -1.0)
+
+        m22 = wp.mat22(1.0, 2.0, 3.0, 4.0)
+        m22 = m22 + m22
+
+        self.assertEqual(m22[1, 1], 8.0)
+        self.assertEqual(str(m22), "[[2.0, 4.0],\n [6.0, 8.0]]")
+
+        t = wp.transform(
+            wp.vec3(1.0, 2.0, 3.0),
+            wp.quat(4.0, 5.0, 6.0, 7.0),
+        )
+        self.assertSequenceEqual(t, (1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
+        self.assertSequenceEqual(t * wp.transform(1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0), (396.0, 432.0, 720.0, 56.0, 70.0, 84.0, -28.0))
+        self.assertSequenceEqual(
+            t * wp.transform((1.0, 2.0, 3.0), (4.0, 5.0, 6.0, 7.0)), (396.0, 432.0, 720.0, 56.0, 70.0, 84.0, -28.0)
+        )
+
+        t = wp.transform()
+        self.assertSequenceEqual(t, (0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 1.0))
+
+        t = wp.transform(p=(1.0, 2.0, 3.0), q=(4.0, 5.0, 6.0, 7.0))
+        self.assertSequenceEqual(t, (1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
+
+        t = wp.transform(q=(4.0, 5.0, 6.0, 7.0), p=(1.0, 2.0, 3.0))
+        self.assertSequenceEqual(t, (1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
+
+        t = wp.transform((1.0, 2.0, 3.0), q=(4.0, 5.0, 6.0, 7.0))
+        self.assertSequenceEqual(t, (1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
+
+        t = wp.transform(p=(1.0, 2.0, 3.0))
+        self.assertSequenceEqual(t, (1.0, 2.0, 3.0, 0.0, 0.0, 0.0, 1.0))
+
+        t = wp.transform(q=(4.0, 5.0, 6.0, 7.0))
+        self.assertSequenceEqual(t, (0.0, 0.0, 0.0, 4.0, 5.0, 6.0, 7.0))
+
+        t = wp.transform((1.0, 2.0, 3.0), (4.0, 5.0, 6.0, 7.0))
+        self.assertSequenceEqual(t, (1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
+
+        t = wp.transform(p=wp.vec3(1.0, 2.0, 3.0), q=wp.quat(4.0, 5.0, 6.0, 7.0))
+        self.assertSequenceEqual(t, (1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
+
+        t = wp.transform(1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0)
+        self.assertSequenceEqual(t, (1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
+
+        t = wp.transform(wp.transform(1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
+        self.assertSequenceEqual(t, (1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
+
+        t = wp.transform(*wp.transform(1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
+        self.assertSequenceEqual(t, (1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0))
+
+        transformf = wp.types.transformation(dtype=float)
+
+        t = wp.transformf((1.0, 2.0, 3.0), (4.0, 5.0, 6.0, 7.0))
+        self.assertSequenceEqual(
+            t + transformf((2.0, 3.0, 4.0), (5.0, 6.0, 7.0, 8.0)),
+            (3.0, 5.0, 7.0, 9.0, 11.0, 13.0, 15.0),
+        )
+        self.assertSequenceEqual(
+            t - transformf((2.0, 3.0, 4.0), (5.0, 6.0, 7.0, 8.0)),
+            (-1.0, -1.0, -1.0, -1.0, -1.0, -1.0, -1.0),
+        )
+
+        f = wp.sin(math.pi * 0.5)
+        self.assertAlmostEqual(f, 1.0, places=3)
+
+        m = wp.mat22(0.0, 0.0, 0.0, 0.0)
+        m += wp.mat22(1.0, 1.0, 1.0, 1.0)
+        assert m == wp.mat22(1.0, 1.0, 1.0, 1.0)
+        m -= wp.mat22(1.0, 1.0, 1.0, 1.0)
+        assert m == wp.mat22(0.0, 0.0, 0.0, 0.0)
+        m = wp.mat22(2.0, 2.0, 2.0, 2.0) - wp.mat22(1.0, 1.0, 1.0, 1.0)
+        assert m == wp.mat22(1.0, 1.0, 1.0, 1.0)
+        m *= 2.0
+        assert m == wp.mat22(2.0, 2.0, 2.0, 2.0)
+        m = m * 2.0
+        assert m == wp.mat22(4.0, 4.0, 4.0, 4.0)
+        m = m / 2.0
+        assert m == wp.mat22(2.0, 2.0, 2.0, 2.0)
+        m /= 2.0
+        assert m == wp.mat22(1.0, 1.0, 1.0, 1.0)
+        m = -m
+        assert m == wp.mat22(-1.0, -1.0, -1.0, -1.0)
+        m = +m
+        assert m == wp.mat22(-1.0, -1.0, -1.0, -1.0)
+        m = m * m
+        assert m == wp.mat22(2.0, 2.0, 2.0, 2.0)
+
+
+    def test_native_function_error_resolution(self):
+        a = wp.mat22f(1.0, 2.0, 3.0, 4.0)
+        b = wp.mat22d(1.0, 2.0, 3.0, 4.0)
+        with self.assertRaisesRegex(
+            RuntimeError,
+            r"^Couldn't find a function 'mul' compatible with " r"the arguments 'mat22f, mat22d'$",
+        ):
+            a * b
+
+
+add_kernel_test(TestFunc, kernel=test_overload_func, name="test_overload_func", dim=1, devices=devices)
+add_function_test(TestFunc, func=test_return_func, name="test_return_func", devices=devices)
+add_kernel_test(TestFunc, kernel=test_override_func, name="test_override_func", dim=1, devices=devices)
+add_function_test(TestFunc, func=test_func_closure_capture, name="test_func_closure_capture", devices=devices)
+add_function_test(TestFunc, func=test_multi_valued_func, name="test_multi_valued_func", devices=devices)
+add_kernel_test(TestFunc, kernel=test_func_defaults, name="test_func_defaults", dim=1, devices=devices)
+add_kernel_test(TestFunc, kernel=test_builtin_shadowing, name="test_builtin_shadowing", dim=1, devices=devices)
 
 
 if __name__ == "__main__":
     wp.build.clear_kernel_cache()
-    _ = register(unittest.TestCase)
-    wp.force_load()
-
     unittest.main(verbosity=2)

warp/tests/test_generics.py

@@ -5,12 +5,13 @@
 # distribution of this software and related documentation without an express
 # license agreement from NVIDIA CORPORATION is strictly prohibited.
 
-import numpy as np
 import unittest
 from typing import Any
 
+import numpy as np
+
 import warp as wp
-from warp.tests.test_base import *
+from warp.tests.unittest_utils import *
 
 wp.init()
 
@@ -463,7 +464,7 @@ def test_type_operator_mispell(test, device):
         i = wp.tid()
         _ = typez(i)(0)
 
-    with test.assertRaisesRegex(RuntimeError, r"Unknown operator 'typez'$"):
+    with test.assertRaisesRegex(RuntimeError, r"Unknown function or operator: 'typez'$"):
         wp.launch(
             kernel,
             dim=1,
@@ -487,81 +488,74 @@ def test_type_attribute_error(test, device):
         )
 
 
-def register(parent):
-    class TestGenerics(parent):
-        pass
-
-    devices = get_test_devices()
-
-    add_kernel_test(TestGenerics, name="test_generic_adder", kernel=test_generic_adder, dim=1, devices=devices)
-    add_kernel_test(TestGenerics, name="test_specialized_func", kernel=test_specialized_func, dim=1, devices=devices)
-
-    add_function_test(TestGenerics, "test_generic_array_kernel", test_generic_array_kernel, devices=devices)
-    add_function_test(TestGenerics, "test_generic_accumulator_kernel", test_generic_accumulator_kernel, devices=devices)
-    add_function_test(TestGenerics, "test_generic_fill", test_generic_fill, devices=devices)
-    add_function_test(TestGenerics, "test_generic_fill_overloads", test_generic_fill_overloads, devices=devices)
-    add_function_test(TestGenerics, "test_generic_transform_kernel", test_generic_transform_kernel, devices=devices)
-    add_function_test(
-        TestGenerics, "test_generic_transform_array_kernel", test_generic_transform_array_kernel, devices=devices
-    )
-    add_function_test(TestGenerics, "test_generic_type_cast", test_generic_type_cast, devices=devices)
-    add_function_test(TestGenerics, "test_generic_type_construction", test_generic_type_construction, devices=devices)
-    add_function_test(
-        TestGenerics, "test_generic_scalar_construction", test_generic_scalar_construction, devices=devices
-    )
-    add_function_test(TestGenerics, "test_generic_type_as_argument", test_generic_type_as_argument, devices=devices)
-
-    foo = Foo()
-    foo.x = 17.0
-    foo.y = 25.0
-    foo.z = 42.0
-
-    bar = Bar()
-    bar.x = wp.vec3(1, 2, 3)
-    bar.y = wp.vec3(10, 20, 30)
-    bar.z = wp.vec3(11, 22, 33)
-
-    add_kernel_test(
-        TestGenerics,
-        name="test_generic_struct_kernel",
-        kernel=test_generic_struct_kernel,
-        dim=1,
-        inputs=[foo],
-        devices=devices,
-    )
-    add_kernel_test(
-        TestGenerics,
-        name="test_generic_struct_kernel",
-        kernel=test_generic_struct_kernel,
-        dim=1,
-        inputs=[bar],
-        devices=devices,
-    )
-
-    add_kernel_test(
-        TestGenerics,
-        name="test_generic_struct_construction_kernel",
-        kernel=test_generic_struct_construction_kernel,
-        dim=1,
-        inputs=[foo],
-        devices=devices,
-    )
-    add_kernel_test(
-        TestGenerics,
-        name="test_generic_struct_construction_kernel",
-        kernel=test_generic_struct_construction_kernel,
-        dim=1,
-        inputs=[bar],
-        devices=devices,
-    )
-
-    add_function_test(TestGenerics, "test_type_operator_mispell", test_type_operator_mispell, devices=devices)
-    add_function_test(TestGenerics, "test_type_attribute_error", test_type_attribute_error, devices=devices)
-
-    return TestGenerics
-
+class TestGenerics(unittest.TestCase):
+    pass
+
+
+devices = get_test_devices()
+
+add_kernel_test(TestGenerics, name="test_generic_adder", kernel=test_generic_adder, dim=1, devices=devices)
+add_kernel_test(TestGenerics, name="test_specialized_func", kernel=test_specialized_func, dim=1, devices=devices)
+
+add_function_test(TestGenerics, "test_generic_array_kernel", test_generic_array_kernel, devices=devices)
+add_function_test(TestGenerics, "test_generic_accumulator_kernel", test_generic_accumulator_kernel, devices=devices)
+add_function_test(TestGenerics, "test_generic_fill", test_generic_fill, devices=devices)
+add_function_test(TestGenerics, "test_generic_fill_overloads", test_generic_fill_overloads, devices=devices)
+add_function_test(TestGenerics, "test_generic_transform_kernel", test_generic_transform_kernel, devices=devices)
+add_function_test(
+    TestGenerics, "test_generic_transform_array_kernel", test_generic_transform_array_kernel, devices=devices
+)
+add_function_test(TestGenerics, "test_generic_type_cast", test_generic_type_cast, devices=devices)
+add_function_test(TestGenerics, "test_generic_type_construction", test_generic_type_construction, devices=devices)
+add_function_test(TestGenerics, "test_generic_scalar_construction", test_generic_scalar_construction, devices=devices)
+add_function_test(TestGenerics, "test_generic_type_as_argument", test_generic_type_as_argument, devices=devices)
+
+foo = Foo()
+foo.x = 17.0
+foo.y = 25.0
+foo.z = 42.0
+
+bar = Bar()
+bar.x = wp.vec3(1, 2, 3)
+bar.y = wp.vec3(10, 20, 30)
+bar.z = wp.vec3(11, 22, 33)
+
+add_kernel_test(
+    TestGenerics,
+    name="test_generic_struct_kernel",
+    kernel=test_generic_struct_kernel,
+    dim=1,
+    inputs=[foo],
+    devices=devices,
+)
+add_kernel_test(
+    TestGenerics,
+    name="test_generic_struct_kernel",
+    kernel=test_generic_struct_kernel,
+    dim=1,
+    inputs=[bar],
+    devices=devices,
+)
+
+add_kernel_test(
+    TestGenerics,
+    name="test_generic_struct_construction_kernel",
+    kernel=test_generic_struct_construction_kernel,
+    dim=1,
+    inputs=[foo],
+    devices=devices,
+)
+add_kernel_test(
+    TestGenerics,
+    name="test_generic_struct_construction_kernel",
+    kernel=test_generic_struct_construction_kernel,
+    dim=1,
+    inputs=[bar],
+    devices=devices,
+)
+add_function_test(TestGenerics, "test_type_operator_mispell", test_type_operator_mispell, devices=devices)
+add_function_test(TestGenerics, "test_type_attribute_error", test_type_attribute_error, devices=devices)
 
 if __name__ == "__main__":
     wp.build.clear_kernel_cache()
-    _ = register(unittest.TestCase)
     unittest.main(verbosity=2)