diff --git a/tests/test_anonymizer.py b/tests/test_anonymizer.py index 503ae6b..32d81f2 100644 --- a/tests/test_anonymizer.py +++ b/tests/test_anonymizer.py @@ -121,16 +121,16 @@ def test_regression(): def test_anonymize_ndarray_one_hot(): x_train = np.array([[23, 0, 1, 165], - [45, 0, 1, 158], - [56, 1, 0, 123], - [67, 0, 1, 154], - [45, 1, 0, 149], - [42, 1, 0, 166], - [73, 0, 1, 172], - [94, 0, 1, 168], - [69, 0, 1, 175], - [24, 1, 0, 181], - [18, 1, 0, 190]]) + [45, 0, 1, 158], + [56, 1, 0, 123], + [67, 0, 1, 154], + [45, 1, 0, 149], + [42, 1, 0, 166], + [73, 0, 1, 172], + [94, 0, 1, 168], + [69, 0, 1, 175], + [24, 1, 0, 181], + [18, 1, 0, 190]]) y_train = np.array([1, 1, 0, 1, 0, 0, 1, 1, 0, 1, 0]) model = DecisionTreeClassifier()