コード例 #1
0
    def test_save_reload_bnn_graph(self):
        m1, _ = self.make_model(self.make_net())
        m1.save(self.TESTFILE)
        m1_loaded = Model()
        m1_loaded.load_graph(self.TESTFILE)
        m1_loaded_edges = set(m1_loaded.components_graph.edges())
        m1_edges = set(m1.components_graph.edges())

        self.assertTrue(set(m1.components) == set(m1_loaded.components))
        self.assertTrue(
            set(m1.components_graph.edges()) == set(
                m1_loaded.components_graph.edges()),
            m1_edges.symmetric_difference(m1_loaded_edges))
        self.assertTrue(
            len(m1_loaded.components.values()) == len(
                set(m1_loaded.components.values())))
        import os
        os.remove(self.TESTFILE)
コード例 #2
0
    def test_save_reload_then_reconcile_bnn_graph(self):
        m1, _ = self.make_model(self.make_net())
        m1.save(self.TESTFILE)
        m1_loaded = Model()
        m1_loaded.load_graph(self.TESTFILE)
        self.assertTrue(set(m1.components) == set(m1_loaded.components))

        m2, _ = self.make_model(self.make_net())
        component_map = mf.models.FactorGraph.reconcile_graphs([m2], m1_loaded)
        self.assertTrue(
            len(component_map.values()) == len(set(component_map.values())),
            "Assert there are only 1:1 mappings.")
        self.assertTrue(len(component_map) == len(m1.components))
        sort_m1 = list(set(map(lambda x: x.uuid, m1.components.values())))
        sort_m1.sort()

        sort_m2 = list(set(map(lambda x: x.uuid, m2.components.values())))
        sort_m2.sort()

        sort_component_map_values = list(set(component_map.values()))
        sort_component_map_values.sort()

        sort_component_map_keys = list(set(component_map.keys()))
        sort_component_map_keys.sort()

        zippy_values = zip(sort_m2, sort_component_map_values)
        zippy_keys = zip(sort_m1, sort_component_map_keys)
        self.assertTrue(
            all([
                m1_item == component_map_item
                for m1_item, component_map_item in zippy_values
            ]))
        self.assertTrue(
            all([
                m2_item == component_map_item
                for m2_item, component_map_item in zippy_keys
            ]))
        import os
        os.remove(self.TESTFILE)