def test_MQ2_write_matrix(self): """ Test MQ² write_matrix function """ data = [[1, 2, 3], [1, 2, 3, 4], 'test'] datafile = os.path.join(TEST_FOLDER, 'test_matrix.csv') self.assertFalse(os.path.exists(datafile)) MQ2.write_matrix(datafile, data) self.assertTrue(os.path.exists(datafile)) matrix = MQ2.read_input_file(datafile, sep=',') self.assertEqual(len(matrix), 3) self.assertEqual(len(matrix[0]), 3) self.assertEqual(len(matrix[1]), 4) self.assertEqual(len(matrix[2]), 1) os.unlink(datafile) self.assertFalse(os.path.exists(datafile))
def test_MQ2_write_matrix(self): """ Test MQ² write_matrix function """ data = [[1,2,3], [1,2,3,4], 'test'] datafile = os.path.join(TEST_FOLDER, 'test_matrix.csv') self.assertFalse(os.path.exists(datafile)) MQ2.write_matrix(datafile, data) self.assertTrue(os.path.exists(datafile)) matrix = MQ2.read_input_file(datafile, sep=',') self.assertEqual(len(matrix), 3) self.assertEqual(len(matrix[0]), 3) self.assertEqual(len(matrix[1]), 4) self.assertEqual(len(matrix[2]), 1) os.unlink(datafile) self.assertFalse(os.path.exists(datafile))