def test_flatten_column_after_pivot(self): """ Test pivot column flattening function """ # single aggregate cases self.assertEqual( proc._flatten_column_after_pivot( aggregates=AGGREGATES_SINGLE, column="idx_nulls", ), "idx_nulls", ) self.assertEqual( proc._flatten_column_after_pivot( aggregates=AGGREGATES_SINGLE, column=1234, ), "1234", ) self.assertEqual( proc._flatten_column_after_pivot( aggregates=AGGREGATES_SINGLE, column=Timestamp("2020-09-29T00:00:00"), ), "2020-09-29 00:00:00", ) self.assertEqual( proc._flatten_column_after_pivot( aggregates=AGGREGATES_SINGLE, column="idx_nulls", ), "idx_nulls", ) self.assertEqual( proc._flatten_column_after_pivot( aggregates=AGGREGATES_SINGLE, column=("idx_nulls", "col1"), ), "col1", ) self.assertEqual( proc._flatten_column_after_pivot( aggregates=AGGREGATES_SINGLE, column=("idx_nulls", "col1", 1234), ), "col1, 1234", ) # Multiple aggregate cases self.assertEqual( proc._flatten_column_after_pivot( aggregates=AGGREGATES_MULTIPLE, column=("idx_nulls", "asc_idx", "col1"), ), "idx_nulls, asc_idx, col1", ) self.assertEqual( proc._flatten_column_after_pivot( aggregates=AGGREGATES_MULTIPLE, column=("idx_nulls", "asc_idx", "col1", 1234), ), "idx_nulls, asc_idx, col1, 1234", )
def test_flatten_column_after_pivot(self): """ Test pivot column flattening function """ # single aggregate cases self.assertEqual( proc._flatten_column_after_pivot( aggregates=AGGREGATES_SINGLE, column="idx_nulls", ), "idx_nulls", ) self.assertEqual( proc._flatten_column_after_pivot( aggregates=AGGREGATES_SINGLE, column=("idx_nulls", "col1"), ), "col1", ) self.assertEqual( proc._flatten_column_after_pivot( aggregates=AGGREGATES_SINGLE, column=("idx_nulls", "col1", "col2"), ), "col1, col2", ) # Multiple aggregate cases self.assertEqual( proc._flatten_column_after_pivot( aggregates=AGGREGATES_MULTIPLE, column=("idx_nulls", "asc_idx", "col1"), ), "idx_nulls, asc_idx, col1", ) self.assertEqual( proc._flatten_column_after_pivot( aggregates=AGGREGATES_MULTIPLE, column=("idx_nulls", "asc_idx", "col1", "col2"), ), "idx_nulls, asc_idx, col1, col2", )
def test_flatten_column_after_pivot(): """ Test pivot column flattening function """ # single aggregate cases assert (_flatten_column_after_pivot( aggregates=AGGREGATES_SINGLE, column="idx_nulls", ) == "idx_nulls") assert (_flatten_column_after_pivot( aggregates=AGGREGATES_SINGLE, column=1234, ) == "1234") assert (_flatten_column_after_pivot( aggregates=AGGREGATES_SINGLE, column=Timestamp("2020-09-29T00:00:00"), ) == "2020-09-29 00:00:00") assert (_flatten_column_after_pivot( aggregates=AGGREGATES_SINGLE, column="idx_nulls", ) == "idx_nulls") assert (_flatten_column_after_pivot( aggregates=AGGREGATES_SINGLE, column=("idx_nulls", "col1"), ) == "col1") assert (_flatten_column_after_pivot( aggregates=AGGREGATES_SINGLE, column=("idx_nulls", "col1", 1234), ) == "col1, 1234") # Multiple aggregate cases assert (_flatten_column_after_pivot( aggregates=AGGREGATES_MULTIPLE, column=("idx_nulls", "asc_idx", "col1"), ) == "idx_nulls, asc_idx, col1") assert (_flatten_column_after_pivot( aggregates=AGGREGATES_MULTIPLE, column=("idx_nulls", "asc_idx", "col1", 1234), ) == "idx_nulls, asc_idx, col1, 1234")