def test_unequal_partial_unordered_duplicates(self):
s = SequenceComparison(1, 2, 2, ordered=False, partial=True)
assert s != (1, 2)
compare(repr(s), expected=(
'\n'
'<SequenceComparison(ordered=False, partial=True)(failed)>\n'
'same:\n'
"[1, 2]\n\n"
'in expected but not actual:\n'
"[2]\n"
'</SequenceComparison(ordered=False, partial=True)>'
))
def test_unequal_nested_unhashable_unordered(self):
s = SequenceComparison({2: True}, {1: True}, {2: True}, {3: True}, ordered=False)
assert s != ({1: True}, {2: True}, {4: True})
compare(repr(s), expected=(
'\n'
'<SequenceComparison(ordered=False, partial=False)(failed)>\n'
'same:\n'
"[{2: True}, {1: True}]\n\n"
'in expected but not actual:\n'
"[{2: True}, {3: True}]\n\n"
'in actual but not expected:\n'
"[{4: True}]\n"
'</SequenceComparison(ordered=False, partial=False)>'
))
def test_unequal_prefix_match_but_partial_false(self):
s = SequenceComparison(1, 2, partial=False)
assert s != (1, 2, 4)
compare(repr(s), expected=(
'\n'
'<SequenceComparison(ordered=True, partial=False)(failed)>\n'
'same:\n'
'[1, 2]\n\n'
'expected:\n'
'[]\n\n'
'actual:\n'
'[4]\n'
'</SequenceComparison(ordered=True, partial=False)>'
))
def test_unequal_generator(self):
s = SequenceComparison(1, 3)
assert s != generator(1, 2)
compare(repr(s), expected=(
'\n'
'<SequenceComparison(ordered=True, partial=False)(failed)>\n'
'same:\n'
"[1]\n\n"
'expected:\n'
'[3]\n\n'
'actual:\n'
'[2]\n'
'</SequenceComparison(ordered=True, partial=False)>'
))
def test_unequal_nested(self):
s = SequenceComparison({1: 'a', 2: 'b'}, [1, 2], recursive=False)
assert s != ({2: 'b', 3: 'c'}, [1, 3])
compare(repr(s), expected=(
'\n'
'<SequenceComparison(ordered=True, partial=False)(failed)>\n'
'same:\n'
"[]\n\n"
'expected:\n'
"[{1: 'a', 2: 'b'}, [1, 2]]\n\n"
'actual:\n'
"[{2: 'b', 3: 'c'}, [1, 3]]\n"
'</SequenceComparison(ordered=True, partial=False)>'
))
def test_unequal_list(self):
s = SequenceComparison(1, 2, 3)
assert s != (1, 2, 4)
compare(repr(s), expected=(
'\n'
'<SequenceComparison(ordered=True, partial=False)(failed)>\n'
'same:\n'
'[1, 2]\n\n'
'expected:\n'
'[3]\n\n'
'actual:\n'
'[4]\n'
'</SequenceComparison(ordered=True, partial=False)>'
))
def test_unequal_full_ordered(self):
s = SequenceComparison(1, 3, 5, ordered=True, partial=False)
assert s != (0, 1, 2, 3, 4)
compare(repr(s), expected=(
'\n'
'<SequenceComparison(ordered=True, partial=False)(failed)>\n'
'same:\n'
'[]\n\n'
'expected:\n'
'[1, 3, 5]\n\n'
'actual:\n'
'[0, 1, 2, 3, 4]\n'
'</SequenceComparison(ordered=True, partial=False)>'
))
def test_unequal_partial_unordered(self):
s = SequenceComparison(1, 3, ordered=False, partial=True)
assert s != (2, 1, 4)
compare(repr(s), expected=(
'\n'
'<SequenceComparison(ordered=False, partial=True)(failed)>\n'
'ignored:\n'
'[2, 4]\n\n'
'same:\n'
"[1]\n\n"
'in expected but not actual:\n'
"[3]\n"
'</SequenceComparison(ordered=False, partial=True)>'
))
def test_unequal_same_but_all_wrong_order(self):
s = SequenceComparison(1, 2, 3)
assert s != (3, 1, 2)
compare(repr(s), expected=(
'\n'
'<SequenceComparison(ordered=True, partial=False)(failed)>\n'
'same:\n'
'[]\n\n'
'expected:\n'
'[1, 2, 3]\n\n'
'actual:\n'
'[3, 1, 2]\n'
'</SequenceComparison(ordered=True, partial=False)>'
))
def test_unequal_partial_ordered_only_one_ignored_recursive(self):
s = SequenceComparison(1, 2, ordered=True, partial=True, recursive=True)
assert s != (2, 1, 4)
compare(repr(s), expected=(
'\n'
'<SequenceComparison(ordered=True, partial=True)(failed)>\n'
'ignored:\n'
'[4]\n\n'
'same:\n'
'[]\n\n'
'expected:\n'
'[1, 2]\n\n'
'actual:\n'
'[2, 1]\n'
'</SequenceComparison(ordered=True, partial=True)>'
))
def test_unequal_partial_ordered_with_prefix(self):
s = SequenceComparison('a', 'b', 1, 2, ordered=True, partial=True)
assert s != ('a', 'b', 2, 1, 4)
compare(repr(s), expected=(
'\n'
'<SequenceComparison(ordered=True, partial=True)(failed)>\n'
'ignored:\n'
'[2, 4]\n\n'
'same:\n'
"['a', 'b', 1]\n\n"
'expected:\n'
'[2]\n\n'
'actual:\n'
'[]\n'
'</SequenceComparison(ordered=True, partial=True)>'
))
def test_unequal_partial_ordered_recursive(self):
s = SequenceComparison(1, 3, 5, ordered=True, partial=True, recursive=True)
assert s != (1, 2, 3, 4, 0)
compare(repr(s), expected=(
'\n'
'<SequenceComparison(ordered=True, partial=True)(failed)>\n'
'ignored:\n'
'[4, 0]\n\n'
'same:\n'
'[1]\n\n'
'expected:\n'
'[3, 5]\n\n'
'actual:\n'
'[2, 3]\n'
'</SequenceComparison(ordered=True, partial=True)>'
))
def test_unequal_nested_recursive(self):
s = SequenceComparison({1: 'a', 2: 'b'}, [1, 2], recursive=True)
assert s != ({2: 'b', 3: 'c'}, [1, 3])
compare(repr(s), expected=(
'\n'
'<SequenceComparison(ordered=True, partial=False)(failed)>\n'
'same:\n'
"[]\n\n"
'expected:\n'
"[{1: 'a', 2: 'b'}, [1, 2]]\n\n"
'actual:\n'
"[{2: 'b', 3: 'c'}, [1, 3]]\n\n"
"While comparing [0]: dict not as expected:\n\n"
"same:\n"
"[2]\n\n"
"in expected but not actual:\n"
"1: 'a'\n\n"
"in actual but not expected:\n"
"3: 'c'\n"
'</SequenceComparison(ordered=True, partial=False)>'
))
def test_no_illegal_features():
"""
Tests whether NoIllegalFeatures works for joins
"""
test_code = cleandoc("""
import pandas as pd
from sklearn.preprocessing import label_binarize, StandardScaler, OneHotEncoder
from sklearn.compose import ColumnTransformer
from sklearn.pipeline import Pipeline
from sklearn.tree import DecisionTreeClassifier
data = pd.DataFrame({'age': [1, 2, 10, 5], 'B': ['cat_a', 'cat_b', 'cat_a', 'cat_c'],
'C': ['cat_a', 'cat_b', 'cat_a', 'cat_c'], 'target': ['no', 'no', 'yes', 'yes']})
column_transformer = ColumnTransformer(transformers=[
('numeric', StandardScaler(), ['age']),
('categorical', OneHotEncoder(sparse=False), ['B', 'C'])
])
income_pipeline = Pipeline([
('features', column_transformer),
('classifier', DecisionTreeClassifier())])
labels = label_binarize(data['target'], classes=['no', 'yes'])
income_pipeline.fit(data, labels)
""")
inspector_result = PipelineInspector \
.on_pipeline_from_string(test_code) \
.add_check(NoIllegalFeatures(['C'])) \
.execute()
check_result = inspector_result.check_to_check_results[NoIllegalFeatures(
['C'])]
# pylint: disable=anomalous-backslash-in-string
expected_result = NoIllegalFeaturesResult(
NoIllegalFeatures(['C']), CheckStatus.FAILURE,
StringComparison("Used illegal columns\: .*"),
SequenceComparison('C', 'age', ordered=False))
compare(check_result, expected_result)
def test_equal_list(self):
s = SequenceComparison(1, 2, 3)
assert s == [1, 2, 3]
def test_equal_nested_unhashable_unordered_partial(self):
s = SequenceComparison({1}, {2}, {2}, ordered=False, partial=True)
assert s == ({2}, {1}, {2}, {3})
def test_equal_ordered_duplicates(self):
s = SequenceComparison(1, 2, 2, ordered=True, partial=True)
assert s == (1, 2, 2, 3)
def test_unequal_bad_type(self):
s = SequenceComparison(1, 3)
assert s != object()
compare(repr(s),
expected="<SequenceComparison(ordered=True, partial=False)(failed)>bad type</>")
def test_equal_partial_ordered(self):
s = SequenceComparison(1, 2, 1, ordered=True, partial=True)
assert s == (1, 1, 2, 1)
def test_repr(self):
compare(repr(SequenceComparison(1, 2, 3)),
expected='<SequenceComparison(ordered=True, partial=False)>1, 2, 3</>')
def test_equal_unordered(self):
s = SequenceComparison(1, 2, 3, ordered=False)
assert s == (1, 3, 2)
def test_equal_partial_unordered(self):
s = SequenceComparison(1, 2, ordered=False, partial=True)
assert s == (2, 1, 4)
def test_equal_generator(self):
s = SequenceComparison(1, 2, 3)
assert s == generator(1, 2, 3)
def test_partial_wrong_type(self):
s = SequenceComparison(partial=True)
assert s != object()
def test_equal_tuple(self):
s = SequenceComparison(1, 2, 3)
assert s == (1, 2, 3)
def test_repr_long(self):
actual = repr(SequenceComparison('a', 'b', 'c'*1000))[:60]
compare(actual,
expected='\n'
"<SequenceComparison(ordered=True, partial=False)>\n'a',\n 'b'")
def test_equal_nested_unhashable_unordered(self):
s = SequenceComparison({1}, {2}, {2}, ordered=False)
assert s == ({2}, {1}, {2})
def test_partial_nothing_specified(self):
s = SequenceComparison(partial=True)
assert s == {}
def test_repr_after_equal(self):
s = SequenceComparison(1, 2, 3)
assert s == (1, 2, 3)
compare(repr(s), expected='<SequenceComparison(ordered=True, partial=False)>1, 2, 3</>')