Implement testing.

NOTE: Have to change demo test, because game_stats.bumps is now more populated.
SaltieRL · Aug 4, 2020 · d847e75 · d847e75 · github-actions · Aug 4, 2020
1 parent 6922200
commit d847e75
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 2 deletions.
diff --git a/carball/tests/stats/bump_test.py b/carball/tests/stats/bump_test.py
@@ -0,0 +1,16 @@
+from carball.tests.utils import get_raw_replays, run_analysis_test_on_replay
+
+from carball.analysis.analysis_manager import AnalysisManager
+
+
+class Test_Bumps:
+    def test_calculate_bumps_correctly(self, replay_cache):
+        def test(analysis: AnalysisManager):
+            proto_game = analysis.get_protobuf_data()
+            count_bumps = 0
+            for i in proto_game.game_stats.bumps:
+                if not i.is_demo:
+                    count_bumps += 1
+            assert count_bumps == 3
+
+        run_analysis_test_on_replay(test, get_raw_replays()["3_BUMPS"], cache=replay_cache)
diff --git a/carball/tests/stats/demo_test.py b/carball/tests/stats/demo_test.py
@@ -7,7 +7,10 @@ class Test_Demos:
     def test_calculate_demos_correctly(self, replay_cache):
         def test(analysis: AnalysisManager):
             proto_game = analysis.get_protobuf_data()
-            bumps = proto_game.game_stats.bumps
-            assert len(bumps) == 1
+            count_demo_bumps = 0
+            for i in proto_game.game_stats.bumps:
+                if i.is_demo:
+                    count_demo_bumps += 1
+            assert count_demo_bumps == 1
 
         run_analysis_test_on_replay(test, get_raw_replays()["1_DEMO"], cache=replay_cache)