def test_multiple_records(self):
"""retrieve_records() multiple records"""
multi_cases = (
# cases 0 and 1, retrieve for the key in case 0
((self.cases[0], self.cases[1]), (self.cases[0],)),
# ... retrieve for key in case 1
((self.cases[0], self.cases[1]), (self.cases[0],)),
# ... retrieve keys for both cases
(
(self.cases[0], self.cases[1]),
(self.cases[0], self.cases[1])
),
# all cases, one key
([case for case in self.cases], (self.cases[1],)),
# all cases, all keys
(
[case for case in self.cases],
[case for case in self.cases]
),
)
for case_dict, recs_to_retrieve, expected in (
self._set_up_multi_cases(multi_cases)):
self.assertEqual(
fastadict.retrieve_records(
case_dict,
recs_to_retrieve
),
expected
)
def test_single_records(self):
"""retrieve_records() single records"""
for case in self.cases:
record_name = case[1].keys()
expected = case[0]
self.assertEqual(
fastadict.retrieve_records(case[1], record_name),
expected
)