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