def test_get_comparison_columns_error(self, table_1, table_2,
include_columns, exclude_columns):
with pytest.raises(ValueError):
assertions.get_comparison_columns(table_1=table_1,
table_2=table_2,
include_columns=include_columns,
exclude_columns=exclude_columns)
def test_get_comparison_columns(self, table_1, table_2, include_columns,
exclude_columns, expected):
actual = assertions.get_comparison_columns(
table_1=table_1,
table_2=table_2,
include_columns=include_columns,
exclude_columns=exclude_columns)
assert list(actual) == expected