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