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")
