def setUp(self): self.matrix_7 = create_matrix(7) self.matrix_13 = create_matrix(13) self.matrix_9 = create_matrix(9) self.sets_7 = set_factory(self.matrix_7[1]) self.sets_13 = set_factory(self.matrix_13[1]) self.sets_9 = set_factory(self.matrix_9[1]) self.network_7 = Network() self.network_7.nodes = assign_set(self.sets_7) self.network_13 = Network() self.network_13.nodes = assign_set(self.sets_13) self.network_9 = Network() self.network_9.nodes = assign_set(self.sets_9)
def testSetFactory(self): self.assertEqual(self.sets_7, ([[1, 2, 3], [1, 4, 5], [1, 6, 7]], [[2, 4, 6], [2, 5, 7]], [[3, 4, 7], [3, 5, 6]])) self.assertEqual(set_factory(create_matrix(6)[1]), ([[1, 2, 3], [1, 4, 5], [1, 6, '_']], [[2, 4, 6], [2, 5, '_']], [[3, 4, '_'], [3, 5, 6]])) self.assertEqual(self.sets_13, ([[1, 2, 3, 4], [1, 5, 6, 7], [1, 8, 9, 10], [1, 11, 12, 13]], [[2, 5, 8, 11], [2, 6, 9, 12], [2, 7, 10, 13]], [[3, 5, 9, 13], [3, 6, 10, 11], [3, 7, 8, 12], [4, 5, 10, 12], [4, 6, 8, 13], [4, 7, 9, 11]])) self.assertEqual((set_factory(create_matrix(12)[1])), ([[1, 2, 3, 4], [1, 5, 6, 7], [1, 8, 9, 10], [1, 11, 12, '_']], [[2, 5, 8, 11], [2, 6, 9, 12], [2, 7, 10, '_']], [[3, 5, 9, '_'], [3, 6, 10, 11], [3, 7, 8, 12], [4, 5, 10, 12], [4, 6, 8, '_'], [4, 7, 9, 11]])) self.assertEqual(set_factory(create_matrix(21)[1]), ([[1, 2, 3, 4, 5], [1, 6, 7, 8, 9], [1, 10, 11, 12, 13], [1, 14, 15, 16, 17], [1, 18, 19, 20, 21]], [[2, 6, 10, 14, 18], [2, 7, 11, 15, 19], [2, 8, 12, 16, 20], [2, 9, 13, 17, 21]], [[3, 6, 11, 16, 21], [3, 7, 12, 17, 18], [3, 8, 13, 14, 19], [3, 9, 10, 15, 20], [4, 6, 12, 14, 20], [4, 7, 13, 15, 21], [4, 8, 10, 16, 18], [4, 9, 11, 17, 19], [5, 6, 13, 16, 19], [5, 7, 10, 17, 20], [5, 8, 11, 14, 21], [5, 9, 12, 15, 18]])) self.assertEqual(set_factory(create_matrix(19)[1]), ([[1, 2, 3, 4, 5], [1, 6, 7, 8, 9], [1, 10, 11, 12, 13], [1, 14, 15, 16, 17], [1, 18, 19, '_', '_']], [[2, 6, 10, 14, 18], [2, 7, 11, 15, 19], [2, 8, 12, 16, '_'], [2, 9, 13, 17, '_']], [[3, 6, 11, 16, '_'], [3, 7, 12, 17, 18], [3, 8, 13, 14, 19], [3, 9, 10, 15, '_'], [4, 6, 12, 14, '_'], [4, 7, 13, 15, '_'], [4, 8, 10, 16, 18], [4, 9, 11, 17, 19], [5, 6, 13, 16, 19], [5, 7, 10, 17, '_'], [5, 8, 11, 14, '_'], [5, 9, 12, 15, 18]]))
def testMatrix(self): self.assertEqual(self.matrix_7, (False, [[4, 5], [6, 7]], [])) self.assertEqual(create_matrix(8), (True, [[5, 6, 7], [8, '_', '_'], ['_', '_', '_']], [(1, 1), (1, 2), (2, 0), (2, 1), (2, 2)])) self.assertEqual(self.matrix_13, (False, [[5, 6, 7], [8, 9, 10], [11, 12, 13]], []))
def testSubstituteDash(self): #self.assertEqual(best_site(matrix(5)[1], 5), '') self.assertEqual(best_site(create_matrix(10)[1], 10, (2, 2)), 3)
def testComputeLinks(self): self.assertEqual(compute_links(set_factory(create_matrix(7)[1])), 14)