Updated tests to check the new pivot value

svaante · Apr 28, 2017 · 5d3f38b · 5d3f38b
1 parent d05c224
commit 5d3f38b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/id3/tests/test_id3.py b/id3/tests/test_id3.py
@@ -29,7 +29,7 @@ def test_info_numerical():
     record = test_splitter._info_numerical(x_numerical_col, y)
     assert_equal(record.split_type, 0)
     assert_equal(record.attribute_counts.size, 2)
-    assert_almost_equal(record.pivot, 2)
+    assert_almost_equal(record.pivot, 3)
     assert_almost_equal(record.info, 0.95, 2)