Add a test

tuda-geo · May 24, 2024 · ae3f283 · ae3f283
1 parent cae9eaf
commit ae3f283
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 1 deletion.
diff --git a/resmda/utils.py b/resmda/utils.py
@@ -60,7 +60,7 @@ def rng(random=None):
 
     """
     if isinstance(random, int):
-        return np.random.default_rng(int)
+        return np.random.default_rng(random)
     elif isinstance(random, np.random.Generator):
         return random
     else:

diff --git a/tests/.keep b/tests/.keep
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -0,0 +1,10 @@
+import numpy as np
+
+from resmda import utils
+
+
+def test_random():
+    assert isinstance(utils.rng(), np.random.Generator)
+    assert isinstance(utils.rng(11), np.random.Generator)
+    rng = np.random.default_rng()
+    assert rng == utils.rng(rng)