test_enum.py 文件源码

def test_expressions_for_values(self):
        self.assertGeneratedOutput(
            """
                enum Bar {
                    TOP = 1 << 0,
                    RIGHT = 1 << 1,
                    BOTTOM = 1 << 2,
                    LEFT = 1 << 3
                };
            """,
            """
            from enum import Enum


            class Bar(Enum):
                TOP = 1
                RIGHT = 2
                BOTTOM = 4
                LEFT = 8
            """
        )