def test_expected_input(self): cats = iTol.gather_categories(self.imap, self.header, self.categories) expected = {'healthy_smoker':iTol.DataCategory({'SID1'},{}), 'pm_smoker':iTol.DataCategory({'SID2'},{}), 'pi_smoker':iTol.DataCategory({'SID3'},{}), 'healthy_ns':iTol.DataCategory({'SID5'},{}), 'pm_ns':iTol.DataCategory({'SID7'},{}), 'pi_ns':iTol.DataCategory({'SID8'},{})} self.assertDictContainsSubset(expected, cats)
def test_no_categories(self): cats = iTol.gather_categories(self.imap, self.header, ['Treatment']) self.assertEqual(len(cats), 1, msg='Result should have only 1 entry') self.assertIn('default', cats, msg='key "default" not found in result')