def test_load_sub_mat(): tmp = """ H WILL-TEST D A test matrix R PMID: 2051488 A Dampier, W T A matrix to test my ability to load in the data J Will dampier's code M rows = ACGT, cols = ACGT 0 -3. 0 -7. -1. 0 -4. 3. 2. 0 """ ID, desc, pmid, author, text, out_dict = CompTools.load_sub_mat(tmp) cor_ID = "WILL-TEST" cor_desc = "A test matrix" cor_pmid = "PMID: 2051488" cor_author = "Dampier, W" cor_text = "A matrix to test my ability to load in the data" cor_dict = {('A', 'A'): 0, ('A', 'C'): -3, ('A', 'G'): -7, ('A', 'T'): -4, ('C', 'A'): -3, ('C', 'C'): 0, ('C', 'G'): -1, ('C', 'T'): 3, ('G', 'A'): -7, ('G', 'C'): -1, ('G', 'G'): 0, ('G', 'T'): 2, ('T', 'A'): -4, ('T', 'C'): 3, ('T', 'G'): 2, ('T', 'T'): 0} yield eq_, cor_ID, ID yield eq_, cor_desc, desc yield eq_, cor_pmid, pmid yield eq_, cor_author, author yield eq_, cor_text, text for key, val in cor_dict.items(): yield eq_, val, out_dict.get(key, None), 'Incorrect value at (%s, %s)' % key