def test_get_entity_by_type(self): learning_cont_yr = LearningContainerYearFactory( academic_year=self.academic_year, additional_entity_1=EntityFactory()) result = learning_cont_yr.get_entity_from_type( entity_container_year_link_type.ADDITIONAL_REQUIREMENT_ENTITY_1) self.assertEqual(result, learning_cont_yr.additional_entity_1) result = learning_cont_yr.get_entity_from_type( entity_container_year_link_type.ADDITIONAL_REQUIREMENT_ENTITY_2) self.assertIsNone(result)