Example #1
0
 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")
Example #2
0
    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")