custom split tests correct var in test

moj-analytical-services · Dec 1, 2020 · ddfda24 · ddfda24
1 parent f39a420
commit ddfda24
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_diagnostics.py b/tests/test_diagnostics.py
@@ -55,5 +55,5 @@ def test_score_hist_splits(spark, gamma_settings_4, params_4, sqlite_con_4):
     res = _splink_score_histogram(df, spark=spark,splits=mysplits)
 
     assert isinstance(res, pd.DataFrame)
-    assert hm.count_rows.count() == 3
-    assert hm.count_rows.sum() == hm.count_rows.cumsum()[2]
+    assert res.count_rows.count() == 3
+    assert res.count_rows.sum() == res.count_rows.cumsum()[2]