test_multipolygon.py

from postgis import MultiPolygon, Polygon

MULTI = (
            (
                ((35, 10), (45, 45), (15, 40), (10, 20), (35, 10)),
                ((20, 30), (35, 35), (30, 20), (20, 30))
            ),
            (
                ((36, 10), (46, 45), (16, 40), (16, 20), (36, 10)),
                ((21, 30), (36, 35), (36, 20), (21, 30))
            ),
        )


def test_multilinestring_geojson():
    multi = MultiPolygon(MULTI)
    assert multi.geojson == {
        "type": "MultiPolygon",
        "coordinates": MULTI
    }


def test_geom_should_compare_with_coords():
    assert MULTI == MultiPolygon(MULTI)


def test_multipolygon_get_item():
    multi = MultiPolygon(MULTI)
    assert multi[0] == Polygon(MULTI[0])


def test_multipolygon_wkt():
    multi = MultiPolygon(MULTI)
    wkt = multi.wkt
    wkt = wkt.replace('.0','')
    wkt = wkt.replace(', ',',')
    assert wkt == 'MULTIPOLYGON(((35 10,45 45,15 40,10 20,35 10),(20 30,35 35,30 20,20 30)),((36 10,46 45,16 40,16 20,36 10),(21 30,36 35,36 20,21 30)))'