Python _flatten_column_after_pivot примеры использования

Пример #1

Файл: pandas_postprocessing_tests.py Проект: PlaidCloud/incubator-superset

    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",
        )

Пример #2

Файл: pandas_postprocessing_tests.py Проект: xvjialing/incubator-superset

    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",
        )

Пример #3

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")