fix dim 0 for random.uniform

Author: AllanZyne

sharpy/random/__init__.py

@@ -1,12 +1,14 @@
 import numpy as np
+import sharpy as sp
 from sharpy import float64
 from sharpy.numpy import fromfunction
 
 def uniform(low, high, size, device='', team=1):
     data = np.random.uniform(low, high, size)
-    # TODO handle 0 dim
-    if isinstance(data, float):
-        return data
+    if len(data.shape) == 0:
+        sp_data = sp.empty(())
+        sp_data[()] = data[()]
+        return sp_data
     return fromfunction(lambda *index: data[index], data.shape, dtype=float64, device=device, team=team)
 
 def rand(*shape, device='', team=1):

test/test_random.py

@@ -23,22 +23,12 @@ def test_random_rand(shape, seed):
     else:
         assert np.allclose(sp.to_numpy(sp_data), np_data)
 
-# @pytest.mark.parametrize("low,high", [(0, 1), (4, 10), (-100, 100)])
-# def test_random_uniform(low, high, shape, seed):
-#     sp.random.seed(seed)
-#     sp_data = sp.random.uniform(low, high, shape)
-
-#     np.random.seed(seed)
-#     np_data = np.random.uniform(low, high, shape)
-
-#     print('np', np_data)
-#     print('sp', sp_data)
+@pytest.mark.parametrize("low,high", [(0, 1), (4, 10), (-100, 100)])
+def test_random_uniform(low, high, shape, seed):
+    sp.random.seed(seed)
+    sp_data = sp.random.uniform(low, high, shape)
 
-#     # 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)
+    np.random.seed(seed)
+    np_data = np.random.uniform(low, high, shape)
 
-sp.init()
-test_random_uniform(0, 1, (), 0)
-sp.fini()
+    assert np.allclose(sp.to_numpy(sp_data), np_data)