| """Tests for address converter.""" |
|
|
| from src import convert_address, batch_convert, ConversionStatus, MappingType |
|
|
|
|
| class TestMergedWard: |
| """Test wards that were merged into new wards.""" |
|
|
| def test_phuc_xa_merged_to_hong_ha(self): |
| result = convert_address("Phường Phúc Xá, Quận Ba Đình, Thành phố Hà Nội") |
| assert result.status == ConversionStatus.SUCCESS |
| assert result.mapping_type == MappingType.MERGED |
| assert result.new.ward == "Phường Hồng Hà" |
| assert result.new.province == "Thành phố Hà Nội" |
|
|
| def test_an_binh_can_tho(self): |
| result = convert_address("Phường An Bình, Quận Ninh Kiều, Thành phố Cần Thơ") |
| assert result.status == ConversionStatus.SUCCESS |
| assert result.new.province == "Thành phố Cần Thơ" |
|
|
|
|
| class TestUnchangedWard: |
| """Test wards that remain unchanged.""" |
|
|
| def test_tan_loc_can_tho(self): |
| result = convert_address("Phường Tân Lộc, Quận Thốt Nốt, Thành phố Cần Thơ") |
| assert result.status == ConversionStatus.SUCCESS |
| assert result.mapping_type == MappingType.UNCHANGED |
| assert result.new.ward == "Phường Tân Lộc" |
|
|
|
|
| class TestRenamedWard: |
| """Test wards that were renamed.""" |
|
|
| def test_long_hoa_renamed(self): |
| result = convert_address("Phường Long Hòa, Quận Bình Thủy, Thành phố Cần Thơ") |
| assert result.status == ConversionStatus.SUCCESS |
| assert result.mapping_type == MappingType.RENAMED |
| assert result.new.ward == "Phường Long Tuyền" |
|
|
|
|
| class TestDividedWard: |
| """Test wards that were split into multiple new wards.""" |
|
|
| def test_divided_selects_default(self): |
| result = convert_address("Xã Tân Thạnh, Huyện Thới Lai, Thành phố Cần Thơ") |
| assert result.status == ConversionStatus.SUCCESS |
| assert result.mapping_type == MappingType.DIVIDED |
|
|
|
|
| class TestAbbreviations: |
| """Test address abbreviation expansion.""" |
|
|
| def test_p_q_tp(self): |
| result = convert_address("P. Phúc Xá, Q. Ba Đình, TP. Hà Nội") |
| assert result.status == ConversionStatus.SUCCESS |
| assert result.new.ward == "Phường Hồng Hà" |
| assert result.new.province == "Thành phố Hà Nội" |
|
|
| def test_tp_shorthand(self): |
| result = convert_address("P.Phúc Xá, Q.Ba Đình, TP.Hà Nội") |
| assert result.status == ConversionStatus.SUCCESS |
| assert result.new.province == "Thành phố Hà Nội" |
|
|
|
|
| class TestPartialAddress: |
| """Test partial addresses (province-only, no ward).""" |
|
|
| def test_province_only(self): |
| result = convert_address("Thành phố Hà Nội") |
| assert result.status == ConversionStatus.PARTIAL |
| assert result.new.province == "Thành phố Hà Nội" |
|
|
| def test_unknown_province(self): |
| result = convert_address("Tỉnh Không Tồn Tại") |
| assert result.status == ConversionStatus.NOT_FOUND |
|
|
|
|
| class TestWithStreet: |
| """Test addresses that include street information.""" |
|
|
| def test_street_preserved(self): |
| result = convert_address("123 Phố Hàng Bông, Phường Phúc Xá, Quận Ba Đình, Thành phố Hà Nội") |
| assert result.status == ConversionStatus.SUCCESS |
| assert "123" in result.converted |
| assert "Phường Hồng Hà" in result.converted |
|
|
|
|
| class TestBatchConvert: |
| """Test batch conversion.""" |
|
|
| def test_batch(self): |
| addresses = [ |
| "Phường Phúc Xá, Quận Ba Đình, Thành phố Hà Nội", |
| "Phường Tân Lộc, Quận Thốt Nốt, Thành phố Cần Thơ", |
| "Tỉnh Không Tồn Tại", |
| ] |
| results = batch_convert(addresses) |
| assert len(results) == 3 |
| assert results[0].status == ConversionStatus.SUCCESS |
| assert results[1].status == ConversionStatus.SUCCESS |
| assert results[2].status == ConversionStatus.NOT_FOUND |
|
|
|
|
| class TestProvinceMapping: |
| """Test province-level conversions (63 -> 34).""" |
|
|
| def test_ha_noi_stays(self): |
| result = convert_address("Thành phố Hà Nội") |
| assert result.new.province == "Thành phố Hà Nội" |
|
|
| def test_merged_province(self): |
| |
| result = convert_address("Tỉnh Hà Giang") |
| assert result.status == ConversionStatus.PARTIAL |
| assert result.new.province |
|
|
