def test_type_of_input_is_not_list(self, data):
columns_list = "Responsavel pelo registro"
with pytest.raises(TypeError):
select_columns(data[0], columns_list)
def test_non_match_one_of_columns(self, data):
columns_list = ["Responsavel pelo registro", "y"]
with pytest.raises(ValueError):
select_columns(data[0], columns_list)
def test_multiple_columns(self, data):
columns_list = ["Responsavel pelo registro", "Data do evento"]
assert select_columns(data[0], columns_list) == utils.select_columns(
data[1], columns_list)
def test_non_match_two_columns(self, data):
columns_list = ["x", "y"]
with pytest.raises(ValueError):
select_columns(data[0], columns_list)
def test_single_column(self, data):
columns_list = ["Responsavel pelo registro"]
assert select_columns(data[0], columns_list) == utils.select_columns(
data[1], columns_list)