tweaks

Author: drbenvincent

causalpy/tests/test_integration_pymc_examples.py

@@ -171,7 +171,7 @@ def test_rd():
 
     Loads data and checks:
     1. data is a dataframe
-    2. pymc_experiments.RegressionDiscontinuity returns correct type
+    2. causalpy.RegressionDiscontinuity returns correct type
     3. the correct number of MCMC chains exists in the posterior inference data
     4. the correct number of MCMC draws exists in the posterior inference data
     """
@@ -200,7 +200,7 @@ def test_rd_bandwidth():
 
     Loads data and checks:
     1. data is a dataframe
-    2. pymc_experiments.RegressionDiscontinuity returns correct type
+    2. causalpy.RegressionDiscontinuity returns correct type
     3. the correct number of MCMC chains exists in the posterior inference data
     4. the correct number of MCMC draws exists in the posterior inference data
     """
@@ -230,7 +230,7 @@ def test_rd_drinking():
 
     Loads data and checks:
     1. data is a dataframe
-    2. pymc_experiments.RegressionDiscontinuity returns correct type
+    2. causalpy.RegressionDiscontinuity returns correct type
     3. the correct number of MCMC chains exists in the posterior inference data
     4. the correct number of MCMC draws exists in the posterior inference data
     """
@@ -290,7 +290,7 @@ def test_rkink():
 
     Loads data and checks:
     1. data is a dataframe
-    2. pymc_experiments.RegressionKink returns correct type
+    2. causalpy.RegressionKink returns correct type
     3. the correct number of MCMC chains exists in the posterior inference data
     4. the correct number of MCMC draws exists in the posterior inference data
     """
@@ -319,7 +319,7 @@ def test_rkink_bandwidth():
 
     Generates synthetic data and checks:
     1. data is a dataframe
-    2. pymc_experiments.RegressionKink returns correct type
+    2. causalpy.RegressionKink returns correct type
     3. the correct number of MCMC chains exists in the posterior inference data
     4. the correct number of MCMC draws exists in the posterior inference data
     """
@@ -349,7 +349,7 @@ def test_its():
 
     Loads data and checks:
     1. data is a dataframe
-    2. pymc_experiments.InterruptedTimeSeries returns correct type
+    2. causalpy.InterruptedTimeSeries returns correct type
     3. the correct number of MCMC chains exists in the posterior inference data
     4. the correct number of MCMC draws exists in the posterior inference data
     """
@@ -385,7 +385,7 @@ def test_its_covid():
 
     Loads data and checks:
     1. data is a dataframe
-    2. pymc_experiments.InterruptedtimeSeries returns correct type
+    2. causalpy.InterruptedtimeSeries returns correct type
     3. the correct number of MCMC chains exists in the posterior inference data
     4. the correct number of MCMC draws exists in the posterior inference data
     """
@@ -422,7 +422,7 @@ def test_sc():
 
     Loads data and checks:
     1. data is a dataframe
-    2. pymc_experiments.SyntheticControl returns correct type
+    2. causalpy.SyntheticControl returns correct type
     3. the correct number of MCMC chains exists in the posterior inference data
     4. the correct number of MCMC draws exists in the posterior inference data
     """
@@ -463,7 +463,7 @@ def test_sc_brexit():
 
     Loads data and checks:
     1. data is a dataframe
-    2. pymc_experiments.SyntheticControl returns correct type
+    2. causalpy.SyntheticControl returns correct type
     3. the correct number of MCMC chains exists in the posterior inference data
     4. the correct number of MCMC draws exists in the posterior inference data
     """
@@ -509,7 +509,7 @@ def test_ancova():
 
     Loads data and checks:
     1. data is a dataframe
-    2. pymc_experiments.PrePostNEGD returns correct type
+    2. causalpy.PrePostNEGD returns correct type
     3. the correct number of MCMC chains exists in the posterior inference data
     4. the correct number of MCMC draws exists in the posterior inference data
     """
@@ -541,7 +541,7 @@ def test_geolift1():
 
     Loads data and checks:
     1. data is a dataframe
-    2. pymc_experiments.SyntheticControl returns correct type
+    2. causalpy.SyntheticControl returns correct type
     3. the correct number of MCMC chains exists in the posterior inference data
     4. the correct number of MCMC draws exists in the posterior inference data
     """

causalpy/tests/test_pymc_experiments.py renamed to causalpy/tests/test_misc.py

@@ -12,7 +12,7 @@
 #   See the License for the specific language governing permissions and
 #   limitations under the License.
 """
-Unit tests for pymc_experiments.py
+Miscellaneous unit tests
 """
 
 import arviz as az
@@ -24,22 +24,6 @@
 sample_kwargs = {"tune": 20, "draws": 20, "chains": 2, "cores": 2}
 
 
-def test_did_summary():
-    """Test that the summary stat function returns a string."""
-    df = cp.load_data("did")
-    result = cp.DifferenceInDifferences(
-        df,
-        formula="y ~ 1 + group*post_treatment",
-        time_variable_name="t",
-        group_variable_name="group",
-        model=cp.pymc_models.LinearRegression(sample_kwargs=sample_kwargs),
-    )
-    round_to = None
-    print(type(result._causal_impact_summary_stat(round_to)))
-    print(result._causal_impact_summary_stat(round_to))
-    assert isinstance(result._causal_impact_summary_stat(round_to), str)
-
-
 def test_regression_kink_gradient_change():
     """Test function to numerically calculate the change in gradient around the kink
     point in regression kink designs"""

docs/source/_static/interrogate_badge.svg

@@ -8,8 +8,13 @@
   :toctree: generated/
 
   data
-  pymc_experiments
   pymc_models
-  skl_experiments
   skl_models
+  experiments
+  expt_diff_in_diff
+  expt_instrumental_variable
+  expt_prepostfit
+  expt_prepostnegd
+  expt_regression_discontinuity
+  expt_regresion_kink
 ```