def test_get_criteria_dict(self): expected_dict = {'fid': 'GDXHsS00004', 'fname': 'Barrett'} criteria_dict = Criteria.get_criteria_dict('GDXHsS00004', 'Barrett') self.assertEqual(expected_dict, criteria_dict, 'dicts are equal') expected_dict = {'fid': 'GDXHsS00004', 'fnotes': {'rsq': '0.1'}, 'fname': 'Barrett'} criteria_dict = Criteria.get_criteria_dict('GDXHsS00004', 'Barrett', {'rsq': '0.1'}) self.assertEqual(expected_dict, criteria_dict, 'dicts are equal')
def test_get_criteria_disease_dict(self): criteria_dict = Criteria.get_criteria_dict('GDXHsS00004', 'Barrett') diseases = ['T1D'] criteria_disease_dict = Criteria.get_criteria_disease_dict(diseases, criteria_dict, {}) expected_dict = {'T1D': [{'fname': 'Barrett', 'fid': 'GDXHsS00004'}]} self.assertEqual(criteria_disease_dict, expected_dict, 'Dict as expected') criteria_disease_dict = Criteria.get_criteria_disease_dict(diseases, criteria_dict, criteria_disease_dict) self.assertEqual(criteria_disease_dict, expected_dict, 'Dict as expected after addding duplicate') criteria_dict = Criteria.get_criteria_dict('GDXHsS00005', 'Catfield') expected_dict = {'T1D': [{'fname': 'Barrett', 'fid': 'GDXHsS00004'}, {'fname': 'Catfield', 'fid': 'GDXHsS00005'}]} criteria_disease_dict = Criteria.get_criteria_disease_dict(diseases, criteria_dict, criteria_disease_dict) self.assertEqual(criteria_disease_dict, expected_dict, 'Dict as expected after adding new') diseases = ['T1D', 'MS'] criteria_disease_dict = Criteria.get_criteria_disease_dict(diseases, criteria_dict, criteria_disease_dict) expected_dict = {'MS': [{'fname': 'Catfield', 'fid': 'GDXHsS00005'}], 'T1D': [{'fname': 'Barrett', 'fid': 'GDXHsS00004'}, {'fname': 'Catfield', 'fid': 'GDXHsS00005'}]} self.assertEqual(criteria_disease_dict, expected_dict, 'Dict as expected after adding diseases')