doc_examples/test_ufunc

Author: acosmicflamingo

numba_cuda/numba/cuda/tests/doc_examples/test_ufunc.py

@@ -8,24 +8,11 @@
 
 
 @skip_on_cudasim("cudasim doesn't support cuda import at non-top-level")
-class TestUFunc(CUDATestCase):
+def test_ufunc():
     """
     Test calling a UFunc
     """
-
-    def setUp(self):
-        # Prevent output from this test showing
-        # up when running the test suite
-        self._captured_stdout = captured_stdout()
-        self._captured_stdout.__enter__()
-        super().setUp()
-
-    def tearDown(self):
-        # No exception type, value, or traceback
-        self._captured_stdout.__exit__(None, None, None)
-        super().tearDown()
-
-    def test_ex_cuda_ufunc_call(self):
+    with captured_stdout():
         # ex_cuda_ufunc.begin
         import numpy as np
         from numba import cuda
@@ -47,7 +34,3 @@ def f(r, x):
         # the sin ufunc inside the kernel, and NumPy's sin ufunc
         np.testing.assert_allclose(r, np.sin(x))
         # ex_cuda_ufunc.end
-
-
-if __name__ == "__main__":
-    unittest.main()