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')
