formatting

Author: max-models

tests/unit/test_cunumpy.py

@@ -1,26 +1,31 @@
 import numpy as np
 import pytest
+
 import cunumpy as xp
 
+
 def test_to_numpy():
     arr = xp.array([1, 2, 3])
     # Even if it's already numpy, to_numpy should work
     arr_np = xp.to_numpy(arr)
     assert isinstance(arr_np, np.ndarray)
     assert np.array_equal(arr_np, [1, 2, 3])
 
+
 def test_to_cunumpy():
     arr = np.array([1, 2, 3])
     arr_xp = xp.to_cunumpy(arr)
     # Backend is numpy in tests usually
     assert isinstance(arr_xp, (np.ndarray, xp.ndarray))
 
+
 def test_get_backend_and_is_gpu_cpu():
     arr = np.array([1, 2, 3])
     assert xp.get_backend(arr) == "numpy"
     assert xp.is_gpu(arr) is False
     assert xp.is_cpu(arr) is True
 
+
 def test_use_backend():
     # Initial backend should be numpy (default) in this test environment
     assert "numpy" in xp.xp.__name__
@@ -32,6 +37,7 @@ def test_use_backend():
 
     assert "numpy" in xp.xp.__name__
 
+
 def test_set_backend():
     # Set to numpy
     xp.set_backend("numpy")
@@ -45,6 +51,7 @@ def test_set_backend():
     arr2 = xp.array([2])
     assert arr2 is not None
 
+
 def test_backend_bools():
     with xp.use_backend("numpy"):
         assert xp.numpy_backend is True

tests/unit/test_cupy.py

@@ -1,21 +1,25 @@
+import numpy as np
 import pytest
+
 import cunumpy as xp
-import numpy as np
+
 
 def test_to_cupy_available():
     try:
         import cupy as cp
     except ImportError:
         pytest.skip("CuPy not installed")
-    
+
     with xp.use_backend("cupy"):
         arr = np.array([1, 2, 3])
         arr_cp = xp.to_cupy(arr)
         assert isinstance(arr_cp, cp.ndarray)
 
+
 def test_to_cupy_not_available():
     try:
         import cupy
+
         pytest.skip("CuPy is installed, cannot test missing cupy error")
     except ImportError:
         pass
@@ -25,22 +29,24 @@ def test_to_cupy_not_available():
         with pytest.raises(ImportError):
             xp.to_cupy(arr)
 
+
 def test_synchronize():
     # Should not crash on any backend
     xp.synchronize()
-    
+
     with xp.use_backend("numpy"):
         xp.synchronize()
-        
+
     with xp.use_backend("cupy"):
         xp.synchronize()
 
+
 def test_xp_array_cupy():
     try:
         import cupy as cp
     except ImportError:
         pytest.skip("CuPy not installed")
-        
+
     with xp.use_backend("cupy"):
         arr = xp.array([1, 2])
         arr *= 2

tests/unit/test_numpy.py

@@ -1,14 +1,17 @@
 import numpy as np
 import pytest
+
 import cunumpy as xp
 
+
 def test_xp_array():
     with xp.use_backend("numpy"):
         arr = xp.array([1, 2])
         arr *= 2
         assert isinstance(arr, np.ndarray)
         assert np.array_equal(arr, [2, 4])
 
+
 def test_numpy_symbols_accessible():
     """All public numpy symbols must be reachable via cunumpy.