def test_parse_genotype_dict_into_table(self): self.assertEqual( table_parser.dict_into_table( ["rs10065172"], {"rs10065172": {"NA18576": "CC"}}), ["\trs10065172", "NA18576\tCC"]) rs_dict = {"rs10050860": {"NA18576": "CC"}, "rs10065172": {"NA18576": "TT"}} self.assertEqual( table_parser.dict_into_table( ["rs10050860", "rs10065172"], rs_dict), ["\trs10050860\trs10065172", "NA18576\tCC\tTT"]) self.assertEqual( table_parser.dict_into_table( ["rs10065172", "rs10050860"], rs_dict), ["\trs10065172\trs10050860", "NA18576\tTT\tCC"])