def test_column_comment(self): class CommentColumn: """hello there hi """ arg = 2 self.assertEqual(get_class_column(CommentColumn), 12)
def test_dataclass_method(self): def wrapper(f): pass @class_decorator class DataclassColumn: @wrapper def func(self): pass self.assertEqual(get_class_column(DataclassColumn), 12)
def test_column_method(self): class FuncColumn: def func(self): pass self.assertEqual(get_class_column(FuncColumn), 12)
def test_column_space(self): class SpaceColumn: arg = 2 self.assertEqual(get_class_column(SpaceColumn), 12)
def test_column_simple(self): class SimpleColumn: arg = 2 self.assertEqual(get_class_column(SimpleColumn), 12)
def test_dataclass(self): @class_decorator class DataclassColumn: arg: int = 5 self.assertEqual(get_class_column(DataclassColumn), 12)