Change type to isinstance checks

Author: twiecki

causalpy/tests/test_synthetic_data.py

@@ -26,7 +26,7 @@ def test_generate_multicell_geolift_data():
     from causalpy.data.simulate_data import generate_multicell_geolift_data
 
     df = generate_multicell_geolift_data()
-    assert type(df) == pd.DataFrame
+    assert isinstance(df, pd.DataFrame)
     assert np.all(df >= 0), "Found negative values in dataset"
 
 
@@ -37,5 +37,5 @@ def test_generate_geolift_data():
     from causalpy.data.simulate_data import generate_geolift_data
 
     df = generate_geolift_data()
-    assert type(df) == pd.DataFrame
+    assert isinstance(df, pd.DataFrame)
     assert np.all(df >= 0), "Found negative values in dataset"