Implement random.seed()

Author: OmarMesqq

integration_tests/test_random.py

@@ -52,6 +52,15 @@ def test_weibullvariate():
     r = random.weibullvariate(-5.6, 1.2)
     print(r)
 
+def test_seed():
+    random.seed(123)
+    t1: f64
+    t1 = random.random()
+    random.seed(321)
+    t2: f64
+    t2 = random.random()
+    assert t1 != t2
+
 def check():
     test_random()
     test_randrange()
@@ -60,5 +69,6 @@ def check():
     test_paretovariate()
     test_expovariate()
     test_weibullvariate()
+    test_seed()
 
 check()

src/libasr/runtime/lfortran_intrinsics.c

@@ -96,6 +96,15 @@ LFORTRAN_API void _lfortran_random_number(int n, double *v)
     }
 }
 
+LFORTRAN_API void _lfortran_init_random(unsigned seed) 
+{
+    if (seed == 0) {
+        srand((unsigned int)clock());
+    } else {
+        srand(seed);
+    }
+}
+
 LFORTRAN_API double _lfortran_random()
 {
     return (rand() / (double) RAND_MAX);

src/libasr/runtime/lfortran_intrinsics.h

@@ -67,6 +67,7 @@ typedef double _Complex double_complex_t;
 
 LFORTRAN_API double _lfortran_sum(int n, double *v);
 LFORTRAN_API void _lfortran_random_number(int n, double *v);
+LFORTRAN_API void _lfortran_init_random(unsigned seed);
 LFORTRAN_API double _lfortran_random();
 LFORTRAN_API int _lfortran_randrange(int lower, int upper);
 LFORTRAN_API int _lfortran_random_int(int lower, int upper);

src/runtime/random.py

@@ -32,6 +32,17 @@ def random() -> f64:
 def _lfortran_random() -> f64:
     pass
 
+def seed(seed: i32) -> None:
+    """
+    Initializes the random number generator.
+    """
+    _lfortran_init_random(seed)
+    return
+
+@ccall
+def _lfortran_init_random(seed: i32) -> None:
+    pass
+
 def randrange(lower: i32, upper: i32) -> i32:
     """
     Return a random integer N such that `lower <= N < upper`.