config run supports cvr & cvr tests

RyanAugust · Feb 3, 2024 · 5ad8bf0 · 5ad8bf0
1 parent c1bb23c
commit 5ad8bf0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/pysimmmulator/simmmulate.py b/src/pysimmmulator/simmmulate.py
@@ -182,3 +182,4 @@ def run_with_config(self):
         self.simulate_baseline(**load_params.cfg['baseline_params'])
         self.simulate_ad_spend(**load_params.cfg["ad_spend_params"])
         self.simulate_media(**load_params.cfg["media_params"])
+        self.simulate_cvr(**load_params.cfg['cvr_params'])
diff --git a/tests/test_sim.py b/tests/test_sim.py
@@ -18,6 +18,13 @@ def test_step3_media():
     sim.simulate_ad_spend(**load_parameters.cfg['ad_spend_params'])
     sim.simulate_media(**load_parameters.cfg['media_params'])
 
+def test_step4_cvr():
+    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'])
+    sim.simulate_cvr(**load_parameters.cfg['cvr_params'])
+
 def test_run_with_config():
     sim = simmmulate.simulate(load_parameters.my_basic_params)
     sim.run_with_config()