int generator and test

synthetic_data/distinct_generators/int_generator.py

@@ -0,0 +1,25 @@
+import numpy as np
+from ..base_generator import BaseGenerator
+from numpy.random import Generator
+
+
+def random_integers(rng: Generator,
+                    min_value: int = -1e6,
+                    max_value: int = 1e6,
+                    num_rows: int = 1
+                    ) -> np.array:
+    """
+    Randomly generates an array of integers between a min and max value
+
+    :param rng: the np rng object used to generate random values
+    :type rng: numpy Generator
+    :param min_value: the minimum integer that can be returned
+    :type min_value: int, optional
+    :param max_value: the maximum integer that can be returned
+    :type max_value: int, optional
+    :param num_rows: the number of rows in np array generated
+    :type num_rows: int, optional
+
+    :return: np array of integers
+    """
+    return rng.integers(min_value, max_value, (num_rows,))

tests/test_int_generator.py

@@ -0,0 +1,34 @@
+import os
+import unittest
+import pandas as pd
+import numpy as np
+
+from synthetic_data.distinct_generators.int_generator import random_integers
+
+
+class TestIntGenerator(unittest.TestCase):
+    def setUp(self):
+        self.rng = np.random.default_rng(12345)     
+
+    def test_return_type(self):
+        result = random_integers(self.rng)
+        self.assertIsInstance(result, np.ndarray)
+        for num in result:
+            self.assertIsInstance(num, np.int64)
+
+    def test_size(self):
+        num_rows = [5,20,100]
+        for nr in num_rows:
+            result = random_integers(self.rng, num_rows=nr)
+            self.assertEqual(result.shape[0], nr)
+        result = random_integers(self.rng)
+        self.assertEqual(result.shape[0], 1)
+
+    def test_values_range(self):
+        ranges = [(-1,1), (-10,10), (-100, 100)]
+        for range in ranges:
+            result = random_integers(self.rng, range[0], range[1])
+            for x in result:
+                self.assertGreaterEqual(x, range[0])
+                self.assertLessEqual(x, range[1])
+