tests coverage for each step

RyanAugust · Jan 31, 2024 · 34413df · 34413df
1 parent d812512
commit 34413df
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tests/test_sim.py b/tests/test_sim.py
@@ -3,6 +3,20 @@
 def test_initiate_sim():
     simmmulate.simulate(load_parameters.my_basic_params)
 
+def test_step1_baseline():
+    sim = simmmulate.simulate(load_parameters.my_basic_params)
+    sim.simulate_baseline(**load_parameters.cfg['baseline_params'])
+
+def test_step2_adspend():
+    sim = simmmulate.simulate(load_parameters.my_basic_params)
+    sim.simulate_baseline(**load_parameters.cfg['baseline_params'])
+    sim.simulate_ad_spend(**load_parameters.cfg['ad_spend_params'])
+
+def test_step3_media():
+    sim = simmmulate.simulate(load_parameters.my_basic_params)
+    sim.simulate_baseline(**load_parameters.cfg['baseline_params'])
+    sim.simulate_ad_spend(**load_parameters.cfg['ad_spend_params'])
+    sim.simulate_media(**load_parameters.cfg['media_params'])
 
 def test_run_with_config():
     sim = simmmulate.simulate(load_parameters.my_basic_params)