class TestFileHandlerCSV(unittest.TestCase): def setUp(self): self.file_valid = "valid_file.csv" with open(self.file_valid, 'w') as file1: file1.write('4,0,0,0,0,0,8,0,5\n') file1.write('0,3,0,0,0,0,0,0,0\n') file1.write('0,0,0,7,0,0,0,0,0\n') file1.write('0,2,0,0,0,0,0,6,0\n') file1.write('0,0,0,0,8,0,4,0,0\n') file1.write('0,0,0,0,1,0,0,0,0\n') file1.write('0,0,0,6,0,3,0,7,0\n') file1.write('5,0,0,2,0,0,0,0,0\n') file1.write('1,0,4,0,0,0,0,0,0\n') self.input_file=FileHandlerCSV(self.file_valid,"r") def tearDown(self): self.input_file.file.close() try: remove(self.file_valid) except: pass def test_CSV_file_returns_the_content_in_a_matrix(self): expected_result=[[4,0,0,0,0,0,8,0,5]\ ,[0,3,0,0,0,0,0,0,0]\ ,[0,0,0,7,0,0,0,0,0]\ ,[0,2,0,0,0,0,0,6,0]\ ,[0,0,0,0,8,0,4,0,0]\ ,[0,0,0,0,1,0,0,0,0]\ ,[0,0,0,6,0,3,0,7,0]\ ,[5,0,0,2,0,0,0,0,0]\ ,[1,0,4,0,0,0,0,0,0]] self.assertEqual(expected_result, self.input_file.import_file())
def setUp(self): self.file_valid = "valid_file.csv" with open(self.file_valid, 'w') as file1: file1.write('4,0,0,0,0,0,8,0,5\n') file1.write('0,3,0,0,0,0,0,0,0\n') file1.write('0,0,0,7,0,0,0,0,0\n') file1.write('0,2,0,0,0,0,0,6,0\n') file1.write('0,0,0,0,8,0,4,0,0\n') file1.write('0,0,0,0,1,0,0,0,0\n') file1.write('0,0,0,6,0,3,0,7,0\n') file1.write('5,0,0,2,0,0,0,0,0\n') file1.write('1,0,4,0,0,0,0,0,0\n') self.input_file=FileHandlerCSV(self.file_valid,"r")