NHSDigital
diff --git a/‎codonPython/__init__.py‎
Lines changed: 0 additions & 4 deletions b/‎codonPython/__init__.py‎
Lines changed: 0 additions & 4 deletions
diff --git a/‎codonPython/nhsNumberGenerator.py‎
Lines changed: 7 additions & 3 deletions b/‎codonPython/nhsNumberGenerator.py‎
Lines changed: 7 additions & 3 deletions
diff --git a/‎codonPython/tests/nhsNumberGenerator_test.py‎
Lines changed: 12 additions & 0 deletions b/‎codonPython/tests/nhsNumberGenerator_test.py‎
Lines changed: 12 additions & 0 deletions
@@ -1,4 +0,0 @@
-from .age_bands import *
-from .dateValidator import *
-from .nhsNumberGenerator import *
-from .tableFromSql import *
@@ -1,7 +1,7 @@
 import random
 
 
-def nhsNumberGenerator(to_generate: int)->list:
+def nhsNumberGenerator(to_generate: int, random_state: int = None)->list:
     """
     Generates random NHS number(s) compliant with modulus 11 checks recorded 
     in the data dictonary.
@@ -11,6 +11,8 @@ def nhsNumberGenerator(to_generate: int)->list:
     ----------
     to_generate : int
         number of NHS numbers to generate
+    random_state : int, default : None
+        Optional seed for random number generation, for testing and reproducability.
 
     Returns
     ----------
@@ -19,11 +21,13 @@ def nhsNumberGenerator(to_generate: int)->list:
 
     Examples
     ---------
-    >>> random.seed(42)
-    >>> nhsNumberGenerator(2)
+    >>> nhsNumberGenerator(2, random_state=42)
     [7865793030, 1933498560]
     """
 
+    if random_state:
+        random.seed(random_state)
+        
     generated = []
     while len(generated) < to_generate:
         # Random 9 digit number starting with non-zero digit
 
@@ -0,0 +1,12 @@
+from codonPython.nhsNumberGenerator import nhsNumberGenerator 
+import numpy as np
+import pytest
+import random
+
+
+@pytest.mark.parametrize("to_generate, random_state, expected", [
+    (3, 42, [7865793030, 1933498560, 7340365060]),
+    (2, 1, [1677604360, 9170772010])
+])
+def test_nhsNumberGenerator_BAU(to_generate, random_state, expected):
+    assert expected == nhsNumberGenerator(to_generate, random_state=random_state)