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)