add test for random

Author: AllanZyne

test/test_random.py

@@ -1,11 +1,24 @@
+import pytest
 import numpy as np
 import sharpy as sp
 
-def test_random_rand(shape, seed):
-    sp.init(False)
+@pytest.fixture(params=[(), (6,), (6, 5), (6, 5, 4)])
+def shape(request):
+    return request.param
+
+@pytest.fixture(params=[0, 42, 66, 890, 1000])
+def seed(request):
+    return request.param
+
 
+def test_random_rand(shape, seed):
     sp.random.seed(seed)
-    a = sp.random.rand(*shape)
-    assert(a.shape == shape)
+    sp_data = sp.random.rand(*shape)
+
+    np.random.seed(seed)
+    np_data = np.random.rand(*shape)
 
-    sp.fini()
+    if isinstance(np_data, float):
+        assert isinstance(sp_data, float) and sp_data == np_data
+    else:
+        assert np.allclose(sp.to_numpy(sp_data), np_data)