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)