def test_add_dataframe(self):
df = self.example_data_A_data[self.one_of_each]
cluster = Cluster('DataFrame')
cluster.add_chain(df)
#check that we have a cluster
self.assertIsInstance(cluster, Cluster)
self.assertItemsEqual('DataFrame', cluster.name)
#check the cluster keys
self.assertItemsEqual(['_'.join(self.one_of_each)], cluster.keys())
#cluster contents
for item in cluster.values():
self.assertIsInstance(item, pd.DataFrame)
def test_add_dataframe(self):
df = self.example_data_A_data[self.one_of_each]
cluster = Cluster('DataFrame')
cluster.add_chain(df)
#check that we have a cluster
self.assertIsInstance(cluster, Cluster)
self.assertCountEqual('DataFrame', cluster.name)
#check the cluster keys
self.assertCountEqual(['_'.join(self.one_of_each)], cluster.keys())
#cluster contents
for item in cluster.values():
self.assertIsInstance(item, pd.DataFrame)
