Código fuente para test.test_translate

import random
import unittest

import mock

from catatom2osm.translate import address_tags, all_tags, building_tags


class TestTranslate(unittest.TestCase):
    def test_all_tags(self):
        tags = {"a": 1, "b": 2, "c": 3}
        fields = []
        for k in list(tags.keys()):
            f = mock.MagicMock()
            f.name.return_value = k
            fields.append(f)
        feat = mock.MagicMock()
        feat.fields.return_value = fields
        feat.side_effect = list(tags.values())
        dest = all_tags(feat)
        for (k, v) in list(tags.items()):
            self.assertTrue(dest[k], str(v))

    def test_address_tags(self):
        self.assertEqual(address_tags({"TN_text": "  ", "postCode": "9"}), {})
        feat = {
            "localId": "000",
            "TN_text": "111",
            "designator": "222",
            "postCode": "",
            "spec": "Parcel",
            "image": "foobar",
        }
        tags = address_tags(feat)
        self.assertEqual(tags["ref"], "000")
        self.assertEqual(tags["addr:street"], "111")
        self.assertEqual(tags["addr:housenumber"], "222")
        self.assertEqual(tags["image"], "foobar")
        self.assertNotIn("addr:postcode", tags)
        self.assertNotIn("entrance", tags)
        feat["spec"] = "Entrance"
        feat["postCode"] = "333"
        tags = address_tags(feat)
        self.assertEqual(tags["entrance"], "yes")
        self.assertEqual(tags["addr:postcode"], "00333")
        feat = {
            "localId": "000",
            "TN_text": "Lugar foo",
            "designator": "",
            "postCode": "",
            "spec": "",
            "image": "foobar",
        }
        tags = address_tags(feat)
        self.assertEqual(tags["addr:place"], "foo")
        self.assertNotIn("addr:street", tags)
        feat = {
            "localId": "000",
            "TN_text": "Plaza foo",
            "designator": "",
            "postCode": "",
            "spec": "",
            "image": "foobar",
        }
        tags = address_tags(feat)
        self.assertEqual(tags["addr:street"], "Plaza foo")
        feat["TN_text"] = "squarePlaza foo"
        tags = address_tags(feat)
        self.assertEqual(tags["addr:place"], "Plaza foo")

    def test_building_tags(self):
        building_values = (
            "residential",
            "barn",
            "industrial",
            "office",
            "retail",
            "public",
        )
        use_values = (
            "1_residential",
            "2_agriculture",
            "3_industrial",
            "4_1_office",
            "4_2_retail",
            "4_3_publicServices",
        )
        feat = {
            "condition": "functional",
            "currentUse": "foobar",
            "nature": None,
            "localId": "foobar",
            "lev_above": 0,
            "lev_below": 0,
            "layer": 1,
            "fixme": "check",
        }
        tags = building_tags(feat)
        self.assertEqual(tags["ref"], "foobar")
        self.assertNotIn("abandoned:building", tags)
        self.assertNotIn("disused:building", tags)
        self.assertEqual(tags["building"], "yes")
        self.assertNotIn("building:levels", tags)
        self.assertNotIn("building:levels:underground", tags)
        self.assertEqual(tags["layer"], "1")
        self.assertEqual(tags["location"], "roof")
        self.assertEqual(tags["fixme"], "check")
        feat["currentUse"] = None
        feat["condition"] = "ruin"
        feat["nature"] = "openAirPool"
        feat["lev_above"] = 1
        feat["lev_below"] = 2
        feat["localId"] = "foobar_part1"
        feat["fixme"] = ""
        tags = building_tags(feat)
        self.assertNotIn("ref", tags)
        self.assertEqual(tags["building"], "ruins")
        self.assertEqual(tags["abandoned:building"], "yes")
        self.assertEqual(tags["leisure"], "swimming_pool")
        self.assertEqual(tags["building:part"], "yes")
        self.assertEqual(tags["building:levels"], "1")
        self.assertEqual(tags["building:levels:underground"], "2")
        self.assertNotIn("fixme", tags)
        use = random.randint(0, len(use_values) - 1)
        feat["currentUse"] = use_values[use]
        feat["condition"] = "declined"
        tags = building_tags(feat)
        self.assertEqual(tags["building"], "yes")
        self.assertEqual(tags["disused:building"], building_values[use])
        feat["lev_above"] = 0
        feat["lev_below"] = 0
        tags = building_tags(feat)
        self.assertEqual(tags["building:part"], "roof")
        self.assertNotIn("building:levels", tags)
        self.assertNotIn("building:levels:underground", tags)

    def test_places(self):
        f1 = {
            "localId": "000",
            "TN_text": "Urbanización A",
            "designator": "1",
            "postCode": "",
            "spec": "Parcel",
            "image": "foobar",
        }
        t = address_tags(f1)
        self.assertNotIn("addr:street", t.keys())
        self.assertEqual(t["addr:place"], "Urbanización A")
        f2 = dict(f1)
        f2["TN_text"] = "urbanización A"
        t = address_tags(f2)
        self.assertNotIn("addr:street", t.keys())
        self.assertEqual(t["addr:place"], "urbanización A")