def setup_retriever(self): self.pattern_measures = {'string pattern': 'Measure(s)'} # pylint: disable=C0301, W0201 self.area = Mock(water_manager='HHNK') # pylint: disable=W0201 retriever = PatternMeasuresRetriever() retriever.retrieve_from_database = Mock(return_value=self.pattern_measures) retriever.retrieve_watertype_group = Mock(return_value='M') return retriever
def test_b(self): """Test the right dict is returned when there is a single ESF pattern""" retriever = PatternMeasuresRetriever() esf_patterns = [0] esf_patterns[0] = self.create_esf_pattern(pattern='XX-------', measures=['dummy measure']) retriever.retrieve_esf_patterns = Mock(return_value=esf_patterns) pattern_measures = retriever.retrieve_from_database('M', 'HHNK') expected_pattern_measures = { esf_patterns[0]: ['dummy measure'] } self.assertEqual(expected_pattern_measures, pattern_measures)
def test_a(self): """Test an empty dict is returned when there are no ESF patterns.""" retriever = PatternMeasuresRetriever() retriever.retrieve_esf_patterns = Mock(return_value=[]) pattern_measures = retriever.retrieve_from_database('M', 'HHNK') self.assertEqual(0, len(pattern_measures))