def test_mq2_read_input_file(self): """ Test MQ² read_input_file function """ matrix = MQ2.read_input_file(os.path.join(TEST_FOLDER, 'mapqtl', 'qtls.exp'), sep=',') self.assertEqual(len(matrix), 5) self.assertEqual(len(matrix[0]), 14)
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))
def test_mq2_read_input_file(self): """ Test MQ² read_input_file function """ matrix = MQ2.read_input_file( os.path.join(TEST_FOLDER, 'mapqtl', 'qtls.exp'), sep=',') self.assertEqual(len(matrix), 5) self.assertEqual(len(matrix[0]), 14)