def setUp(self): self.heuristic = SortHeuristic()
class TestSortHeuristicMatching(unittest.TestCase): def setUp(self): self.heuristic = SortHeuristic() def test_empty_sequence(self): self.heuristic.preprocess([]) self.assertEqual(self.heuristic.query(7), []) # testando se ha lixo entre pesquisas self.assertEqual(self.heuristic.query(10), []) def test_sequence_size_one(self): self.heuristic.preprocess([3]) self.assertEqual(self.heuristic.query(10), []) def test_sequence_size_two(self): self.heuristic.preprocess([3, 1]) self.assertEqual(self.heuristic.query(10), []) # vazio pela ordem dos numeros self.assertEqual(self.heuristic.query(0), []) self.heuristic.preprocess([1, 3]) self.assertEqual(self.heuristic.query(10), []) self.assertEqual(self.heuristic.query(0), [(1, 3)]) def test_equal_values_sequence(self): self.heuristic.preprocess([1]*100) self.assertEqual(self.heuristic.query(1), []) self.assertEqual(len(self.heuristic.query(0)), (100*99)/2) def test_arithmetic_sequence(self): # sequencia de 1 ate 100 self.heuristic.preprocess(range(1, 101)) self.assertEqual(self.heuristic.query(99), [(1, 100)]) # todos os pares tem diferenca maior ou igual a 1 self.assertEqual(len(self.heuristic.query(1)), (100 * 99)/2) answers98 = self.heuristic.query(98) self.assertEqual(len(answers98), 3) self.assertTrue((1, 100) in answers98) self.assertTrue((1, 99) in answers98) self.assertTrue((2, 100) in answers98)
class TestSortHeuristicMatching(unittest.TestCase):
def setUp(self):
self.heuristic = SortHeuristic()
def test_empty_sequence(self):
self.heuristic.preprocess([])
self.assertEqual(self.heuristic.query(7), [])
# testando se ha lixo entre pesquisas
self.assertEqual(self.heuristic.query(10), [])
def test_sequence_size_one(self):
self.heuristic.preprocess([3])
self.assertEqual(self.heuristic.query(10), [])
def test_sequence_size_two(self):
self.heuristic.preprocess([3, 1])
self.assertEqual(self.heuristic.query(10), [])
# vazio pela ordem dos numeros
self.assertEqual(self.heuristic.query(0), [])
self.heuristic.preprocess([1, 3])
self.assertEqual(self.heuristic.query(10), [])
self.assertEqual(self.heuristic.query(0), [(1, 3)])
def test_equal_values_sequence(self):
self.heuristic.preprocess([1] * 100)
self.assertEqual(self.heuristic.query(1), [])
self.assertEqual(len(self.heuristic.query(0)), (100 * 99) / 2)
def test_arithmetic_sequence(self):
# sequencia de 1 ate 100
self.heuristic.preprocess(range(1, 101))
self.assertEqual(self.heuristic.query(99), [(1, 100)])
# todos os pares tem diferenca maior ou igual a 1
self.assertEqual(len(self.heuristic.query(1)), (100 * 99) / 2)
answers98 = self.heuristic.query(98)
self.assertEqual(len(answers98), 3)
self.assertTrue((1, 100) in answers98)
self.assertTrue((1, 99) in answers98)
self.assertTrue((2, 100) in answers98)
def setUp(self):
self.heuristic = SortHeuristic()
def get_alg(self, alg):
if alg == "SortHeuristic":
from heuristics.sort_heuristic import SortHeuristic
return SortHeuristic()