class ParserTest(unittest.TestCase):
def setUp(self):
self.test_input_one = 'tests/input_files/test_input_one.in'
self.test_input_two = 'tests/input_files/test_input_two.in'
self.test_cancer = 'tests/input_files/test_cancer.in'
self.test_parser_one = Parser(self.test_input_one)
self.test_parser_two = Parser(self.test_input_two)
self.test_parser_cancer = Parser(self.test_cancer)
def test_matrix_generator(self):
'''
Tests three input files for a correctly generated matrix.
The third test checks if the validator is working correctly,
so we check if a ValueError is appropriately raised.
'''
observed_one = self.test_parser_one.generate_matrix()
observed_two = self.test_parser_two.generate_matrix()
expected_one = [[0, 0, 1, 0],
[0, 0, 0, 1],
[0, 1, 0, 1],
[1, 0, 0, 0]]
expected_two = [[0 for i in range(15)] for j in range(15)]
self.assertEqual(observed_one, expected_one)
self.assertEqual(observed_two, expected_two)
with self.assertRaises(ValueError):
self.test_parser_cancer.generate_matrix()
def setUp(self):
self.test_input_one = 'tests/input_files/test_input_one.in'
self.test_input_two = 'tests/input_files/test_input_two.in'
self.test_cancer = 'tests/input_files/test_cancer.in'
self.test_parser_one = Parser(self.test_input_one)
self.test_parser_two = Parser(self.test_input_two)
self.test_parser_cancer = Parser(self.test_cancer)
