create simple test cases

.github/workflows/unittest.yml

@@ -0,0 +1,20 @@
+name: Python unittest
+
+on: [push]
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v4
+    - name: Set up Python
+      uses: actions/setup-python@v4
+      with:
+        python-version: '3.11'
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        pip install -r requirements.txt
+    - name: Running tests
+      run: |
+        python -m unittest discover -p '*_test.py'

cloud_function/replay_from_bigquery.py

@@ -51,7 +51,7 @@ def is_before_simulation(self, time: datetime):
         return self.get_simulation_time(time) < self.simulation_start_time
 
     def is_after_simulation(self, time: datetime):
-        return self.get_simulation_time(time) > self.simulation_end_time
+        return self.get_simulation_time(time) >= self.simulation_end_time
 
     def get_simulation_time(self, time: datetime):
         return self.simulation_start_time + (time - self.real_start_time) * self.speed

cloud_function/replay_from_bigquery_test.py

@@ -0,0 +1,40 @@
+import unittest
+from datetime import datetime
+
+from cloud_function.replay_from_bigquery import TimeConfig
+
+
+class TestTimeConfig(unittest.TestCase):
+    config = TimeConfig(
+        real_start_time=datetime.fromtimestamp(4000).isoformat(),
+        simulation_start_time=datetime.fromtimestamp(10000).isoformat(),
+        simulation_end_time=datetime.fromtimestamp(20000).isoformat(),
+        speed=2,
+        scheduler_name=""
+    )
+
+    def test_is_before_simulation(self):
+        self.assertEqual(self.config.is_before_simulation(datetime.fromtimestamp(12000)), False)
+        self.assertEqual(self.config.is_before_simulation(datetime.fromtimestamp(10000)), False)
+        self.assertEqual(self.config.is_before_simulation(datetime.fromtimestamp(6000)), False)
+        self.assertEqual(self.config.is_before_simulation(datetime.fromtimestamp(4000)), False)
+        self.assertEqual(self.config.is_before_simulation(datetime.fromtimestamp(2000)), True)
+        self.assertEqual(self.config.is_before_simulation(datetime.fromtimestamp(3999)), True)
+
+    def test_is_after_simulation(self):
+        self.assertEqual(self.config.is_after_simulation(datetime.fromtimestamp(12000)), True)
+        self.assertEqual(self.config.is_after_simulation(datetime.fromtimestamp(10000)), True)
+        self.assertEqual(self.config.is_after_simulation(datetime.fromtimestamp(9000)), True)
+        self.assertEqual(self.config.is_after_simulation(datetime.fromtimestamp(8999)), False)
+        self.assertEqual(self.config.is_after_simulation(datetime.fromtimestamp(2000)), False)
+        self.assertEqual(self.config.is_after_simulation(datetime.fromtimestamp(3999)), False)
+
+    def test_get_simulation_time(self):
+        self.assertEqual(self.config.get_simulation_time(datetime.fromtimestamp(4000)), datetime.fromtimestamp(10000))
+        self.assertEqual(self.config.get_simulation_time(datetime.fromtimestamp(4001)), datetime.fromtimestamp(10002))
+        self.assertEqual(self.config.get_simulation_time(datetime.fromtimestamp(6000)), datetime.fromtimestamp(14000))
+        self.assertEqual(self.config.get_simulation_time(datetime.fromtimestamp(9000)), datetime.fromtimestamp(20000))
+
+
+if __name__ == '__main__':
+    unittest.main()