Added test_binarizer

EpistasisLab · Apr 16, 2016 · b359df2 · b359df2
1 parent 3834b13
commit b359df2
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tests.py b/tests.py
@@ -243,3 +243,7 @@ def test_div():
         tpot_obj = TPOT()
         assert tpot_obj._div(5,0)==0
 
+def test_binarizer():
+        tpot_obj = TPOT()
+
+        assert np.array_equal(tpot_obj._binarizer(training_testing_data.ix[:,-3:],0),training_testing_data.ix[:,-3:])