def test_accepts_valid_partitions(self):
columns, dataframe = self.setup_df()
parq.check_partitions(columns, dataframe)
def test_reject_non_column_partitions(self):
columns, dataframe = self.setup_df()
with pytest.raises(ValueError):
parq.check_partitions(('banana', ), dataframe)
def test_accepts_valid_partitions(self):
dataframe = setup_grouped_dataframe()
parq.check_partitions(dataframe.columns, dataframe)
