def test_site_disease_codes(self): ''' Test getting all diseases on the site ''' from disease.utils import Disease (main, other) = Disease.get_site_disease_codes() self.assertEqual(12, len(main), "12 main diseases found ") self.assertEqual(7, len(other), "7 other diseases found when searching for other diseases")