fix unit tests

Author: SvenKlaassen

doubleml/tests/test_dml_data.py

@@ -24,7 +24,7 @@ def test_doubleml_basedata():
     dummy_dml_data = DummyDataClass(pd.DataFrame(np.zeros((100, 10))))
     assert dummy_dml_data.d_cols[0] == 'theta'
     assert dummy_dml_data.n_treat == 1
-
+    assert dummy_dml_data.n_coefs == 1
 
 @pytest.fixture(scope="module")
 def dml_data_fixture(generate_data1):

doubleml/tests/test_doubleml_exceptions.py

@@ -181,6 +181,7 @@ def test_doubleml_exception_data():
         # no instrument Z for LPQ
         _ = DoubleMLLPQ(DoubleMLData(df_iivm, 'y', 'd', x_cols=['z']),
                         LogisticRegression(), LogisticRegression(), treatment=1)
+    with pytest.raises(ValueError, match=msg):
         # non-binary Z for LPQ
         _ = DoubleMLLPQ(DoubleMLData(df_iivm, 'y', 'd', z_cols=['z']),
                         LogisticRegression(), LogisticRegression(), treatment=1)