def check_postcode_found(self, postcode, expected_result): self.m.return_value = [self.valid_entry_one_letter, self.valid_entry_two_letters] with patch('geo.open', self.m, create=True): g = Geo() e, n = g.getEastingAndNorthing(postcode) assert_equal((e, n), expected_result) postcode_dir = url_for('static', filename='postcodes') filename = re.match('([a-z]{1,2})[0-9]', postcode.replace(' ', '')).group(1) filename = os.path.join(postcode_dir, filename + '.csv') self.m.assert_called_once_with(filename)
def check_postcode_not_found(self, file_content, postcode): self.m.return_value = file_content with patch('geo.open', self.m, create=True): g = Geo() with assert_raises(PostcodeNotFoundError): g.getEastingAndNorthing(postcode)
def check_invalid_postcode(self, postcode): with patch('geo.open', self.m, create=True): g = Geo() with assert_raises(PostcodeMalformedError): g.getEastingAndNorthing(postcode)