def setUp(self):
self.data = [[1, 'block', 1.4],
[2, 'community', 6.0],
[2, 'test', 5],
[3, 'ok', 7.9],
[None, '', 2.0]]
self.df = pd.DataFrame(self.data, columns=['col1', 'col2', 'col3'])
self.dfs = DataFrameSummary(self.df)
class TestDataFrameSummary(TestCase):
def setUp(self):
self.data = [[1, 'block', 1.4],
[2, 'community', 6.0],
[2, 'test', 5],
[3, 'ok', 7.9],
[None, '', 2.0]]
self.df = pd.DataFrame(self.data, columns=['col1', 'col2', 'col3'])
self.dfs = DataFrameSummary(self.df)
def test_setup(self):
assert isinstance(self.dfs, pd.DataFrame) is True
def test_get_column_types(self):
col_types_df = self.dfs.get_column_types()
assert isinstance(col_types_df, pd.DataFrame) is True
assert col_types_df.columns.tolist() == ['column_name', 'value', 'variable']