def test_invalid_data_bad_header(self): with self.assertRaises(GeonamesParseError): parse_geonames_data(iter(self.invalid_data_bad_header))
def test_invalid_data_field_length_update(self): parse_geonames_data(iter(self.valid_data)) with self.assertRaises(GeonamesParseError): parse_geonames_data(iter(self.invalid_data_invalid_field_length))
def test_invalid_data_field_length_update( self, valid_data, invalid_data_invalid_field_length): parse_geonames_data(iter(valid_data)) with pytest.raises(GeonamesParseError): parse_geonames_data(iter(invalid_data_invalid_field_length))
def test_valid_data(self): parse_geonames_data(iter(self.valid_data)) self.assertEqual(Country.objects.count(), 1)
def test_invalid_data_bad_header(self, invalid_data_bad_header): with pytest.raises(GeonamesParseError): parse_geonames_data(iter(invalid_data_bad_header))
def test_valid_data(self, valid_data): parse_geonames_data(iter(valid_data)) assert Country.objects.count() == 1