Пример #1
0
    def test_size_10_graph(self):
        graph = Graph(10)

        self.add_rays(graph)

        colors_1 = self.heuristics.color_graph(graph)
        colors_2 = self.exact.get_colors_by_graph(graph)

        self.assertEqual(10, len(colors_1))
        self.assertEqual(2, len(set(colors_1.values())))

        self.assertEqual(10, len(colors_2))
        self.assertEqual(2, len(set(colors_2)))

        self.add_circuit(graph)

        self.exact = Exact({})
        colors_1 = self.heuristics.color_graph(graph)
        colors_2 = self.exact.get_colors_by_graph(graph)

        self.assertEqual(10, len(colors_1))
        self.assertEqual(3, len(set(colors_1.values())))

        self.assertEqual(10, len(colors_2))
        self.assertEqual(3, len(set(colors_2)))
Пример #2
0
    def test_size_10_graph(self):
        graph = Graph(10)

        self.add_rays(graph)

        colors_1 = self.heuristics.color_graph(graph)
        colors_2 = self.exact.get_colors_by_graph(graph)

        self.assertEqual(10, len(colors_1))
        self.assertEqual(2, len(set(colors_1.values())))

        self.assertEqual(10, len(colors_2))
        self.assertEqual(2, len(set(colors_2)))

        self.add_circuit(graph)

        self.exact = Exact({})
        colors_1 = self.heuristics.color_graph(graph)
        colors_2 = self.exact.get_colors_by_graph(graph)

        self.assertEqual(10, len(colors_1))
        self.assertEqual(3, len(set(colors_1.values())))

        self.assertEqual(10, len(colors_2))
        self.assertEqual(3, len(set(colors_2)))
Пример #3
0
def main():
    if len(sys.argv) != 3:
        print_usage()
        return
    shape_file_path = sys.argv[1]
    out_file_path = sys.argv[2]

    print "Loading countries..."
    countries = load_countries(shape_file_path)
    print str.format("Loaded {0} countries", len(countries))

    print "Building graph..."
    graph = graph_builder.Builder.build_country_graph_from_countries(countries)
    print str.format("Loaded graph of {0} vertices", graph.get_vertices_count())

    print "Coloring graph..."
    if len(countries) < 15:
        print "Using exact algorithm"
        colors_map = Exact.Exact().get_colors_by_graph(graph)
    else:
        print "Using approximation algorithm"
        colors_map = Heuristics.Heurstics().color_graph(graph)

    print "Dumping results..."
    Serializer.Serializer.dump_to_file(countries, colors_map, out_file_path)
Пример #4
0
 def setUp(self):
     self.heuristics = Heurstics
     self.exact = Exact({})
Пример #5
0
class TestSpeed(unittest.TestCase):
    def setUp(self):
        self.heuristics = Heurstics
        self.exact = Exact({})

    def add_rays(self, graph):
        size = graph.get_vertices_count()
        for i in xrange(1, size):
            graph.add_edge(0, i)

    def add_circuit(self, graph):
        size = graph.get_vertices_count()
        for i in xrange(size):
            graph.add_edge(i, (i + 1) % size)

    def test_size_10_graph(self):
        graph = Graph(10)

        self.add_rays(graph)

        colors_1 = self.heuristics.color_graph(graph)
        colors_2 = self.exact.get_colors_by_graph(graph)

        self.assertEqual(10, len(colors_1))
        self.assertEqual(2, len(set(colors_1.values())))

        self.assertEqual(10, len(colors_2))
        self.assertEqual(2, len(set(colors_2)))

        self.add_circuit(graph)

        self.exact = Exact({})
        colors_1 = self.heuristics.color_graph(graph)
        colors_2 = self.exact.get_colors_by_graph(graph)

        self.assertEqual(10, len(colors_1))
        self.assertEqual(3, len(set(colors_1.values())))

        self.assertEqual(10, len(colors_2))
        self.assertEqual(3, len(set(colors_2)))

    def test_size_25(self):
        graph = Graph(25)

        self.add_rays(graph)

        colors_1 = self.heuristics.color_graph(graph)
        #why toooo long
        #colors_2 = self.exact.get_colors_by_graph(graph)

        self.assertEqual(25, len(colors_1))
        self.assertEqual(2, len(set(colors_1.values())))

        #self.assertEqual(25, len(colors_2))
        #self.assertEqual(3, len(set(colors_2)))

        self.add_circuit(graph)

        colors_1 = self.heuristics.color_graph(graph)
        #why toooo long
        #colors_2 = self.exact.get_colors_by_graph(graph)

        self.assertEqual(25, len(colors_1))
        self.assertEqual(3, len(set(colors_1.values())))

        #self.assertEqual(25, len(colors_2))
        #self.assertEqual(3, len(set(colors_2)))

    def test_size_100_graph(self):
        graph = Graph(100)

        self.add_rays(graph)

        colors = self.heuristics.color_graph(graph)

        self.assertEqual(100, len(colors))
        self.assertEqual(2, len(set(colors.values())))

        self.add_circuit(graph)

        colors = self.heuristics.color_graph(graph)

        self.assertEqual(100, len(colors))
        self.assertEqual(3, len(set(colors.values())))

    def test_size_300_graph(self):
        graph = Graph(300)

        self.add_rays(graph)

        colors = self.heuristics.color_graph(graph)

        self.assertEqual(300, len(colors))
        self.assertEqual(2, len(set(colors.values())))

        self.add_circuit(graph)

        colors = self.heuristics.color_graph(graph)
        self.assertEqual(300, len(colors))
        self.assertEqual(3, len(set(colors.values())))
Пример #6
0
class TestSpeed(unittest.TestCase):
    def setUp(self):
        self.heuristics = Heurstics
        self.exact = Exact({})

    def add_rays(self, graph):
        size = graph.get_vertices_count()
        for i in xrange(1, size):
            graph.add_edge(0, i)

    def add_circuit(self, graph):
        size = graph.get_vertices_count()
        for i in xrange(size):
            graph.add_edge(i, (i+1) % size)

    def test_size_10_graph(self):
        graph = Graph(10)

        self.add_rays(graph)

        colors_1 = self.heuristics.color_graph(graph)
        colors_2 = self.exact.get_colors_by_graph(graph)

        self.assertEqual(10, len(colors_1))
        self.assertEqual(2, len(set(colors_1.values())))

        self.assertEqual(10, len(colors_2))
        self.assertEqual(2, len(set(colors_2)))

        self.add_circuit(graph)

        self.exact = Exact({})
        colors_1 = self.heuristics.color_graph(graph)
        colors_2 = self.exact.get_colors_by_graph(graph)

        self.assertEqual(10, len(colors_1))
        self.assertEqual(3, len(set(colors_1.values())))

        self.assertEqual(10, len(colors_2))
        self.assertEqual(3, len(set(colors_2)))

    def test_size_25(self):
        graph = Graph(25)

        self.add_rays(graph)

        colors_1 = self.heuristics.color_graph(graph)
        #why toooo long
        #colors_2 = self.exact.get_colors_by_graph(graph)

        self.assertEqual(25, len(colors_1))
        self.assertEqual(2, len(set(colors_1.values())))

        #self.assertEqual(25, len(colors_2))
        #self.assertEqual(3, len(set(colors_2)))

        self.add_circuit(graph)

        colors_1 = self.heuristics.color_graph(graph)
        #why toooo long
        #colors_2 = self.exact.get_colors_by_graph(graph)

        self.assertEqual(25, len(colors_1))
        self.assertEqual(3, len(set(colors_1.values())))

        #self.assertEqual(25, len(colors_2))
        #self.assertEqual(3, len(set(colors_2)))

    def test_size_100_graph(self):
        graph = Graph(100)

        self.add_rays(graph)

        colors = self.heuristics.color_graph(graph)

        self.assertEqual(100, len(colors))
        self.assertEqual(2, len(set(colors.values())))

        self.add_circuit(graph)

        colors = self.heuristics.color_graph(graph)

        self.assertEqual(100, len(colors))
        self.assertEqual(3, len(set(colors.values())))

    def test_size_300_graph(self):
        graph = Graph(300)

        self.add_rays(graph)

        colors = self.heuristics.color_graph(graph)

        self.assertEqual(300, len(colors))
        self.assertEqual(2, len(set(colors.values())))

        self.add_circuit(graph)

        colors = self.heuristics.color_graph(graph)
        self.assertEqual(300, len(colors))
        self.assertEqual(3, len(set(colors.values())))
Пример #7
0
 def setUp(self):
     self.heuristics = Heurstics
     self.exact = Exact({})
Пример #8
0
 def setUp(self):
     self.exact = Exact.Exact({})