def test_load_and_save_to_disk(self): clusters =(Cluster(16,[16]), Cluster(4,[4,5,6,7,8]), Cluster(0,[0,1,2,3]), Cluster(9,[9,10,11,12,13,14,15])) clustering = Clustering(clusters) before_saving_elements = clustering.get_all_clustered_elements() clustering.save_to_disk("data/saved_clustering_for_test") loaded_clustering = Clustering.load_from_disk("data/saved_clustering_for_test") after_saving_elements = loaded_clustering.get_all_clustered_elements() self.assertItemsEqual(before_saving_elements, after_saving_elements) os.system("rm data/saved_clustering_for_test")
def test_load_and_save_to_disk(self): clusters = (Cluster(16, [16]), Cluster(4, [4, 5, 6, 7, 8]), Cluster(0, [0, 1, 2, 3]), Cluster(9, [9, 10, 11, 12, 13, 14, 15])) clustering = Clustering(clusters) before_saving_elements = clustering.get_all_clustered_elements() clustering.save_to_disk( os.path.join(test_data.__path__[0], "saved_clustering_for_test")) loaded_clustering = Clustering.load_from_disk( os.path.join(test_data.__path__[0], "saved_clustering_for_test")) after_saving_elements = loaded_clustering.get_all_clustered_elements() self.assertItemsEqual(before_saving_elements, after_saving_elements) os.system("rm data/saved_clustering_for_test")