63 lines
3.0 KiB
Python
63 lines
3.0 KiB
Python
from shapely.geometry import LineString, Point, box
|
||
from shapely.ops import unary_union
|
||
|
||
from leonida.park import Street, World, apply_street
|
||
|
||
STREET = Street(start=(150.0, 660.0), end=(150.0, 320.0), width=12.0,
|
||
remove_buildings=((155.0, 615.0),),
|
||
alley_cut=((90.0, 639.0), (110.0, 639.0), (110.0, 341.0), (90.0, 341.0)))
|
||
|
||
|
||
def world():
|
||
"""Querstraßen bei y 640–660 und 320–340, dazwischen eine Gasse bei x 95–105."""
|
||
removed = box(140, 600, 170, 630)
|
||
return World(
|
||
land=[box(-100, 200, 300, 800)],
|
||
surfaces=[(box(0, 340, 95, 640), "urban"), (box(105, 340, 200, 640).difference(removed), "urban")],
|
||
roads=[(unary_union([box(0, 640, 200, 660), box(0, 320, 200, 340), box(95, 340, 105, 640)]), False)],
|
||
centerlines=[LineString([(0, 650), (200, 650)]), LineString([(0, 330), (200, 330)]),
|
||
LineString([(100, 650), (100, 330)])],
|
||
buildings=[(removed, {"height": 20.9, "source": "district"}, False),
|
||
(box(20, 400, 60, 450), {"height": 10.8, "source": "district"}, False),
|
||
(box(84, 450, 93, 470), {"height": 8.0, "source": "district"}, False)],
|
||
palms=[(150.0, 500.0), (60.0, 500.0), (140.0, 500.0)],
|
||
)
|
||
|
||
|
||
def on_road(result, x, y):
|
||
return any(poly.contains(Point(x, y)) for poly, _ in result.roads)
|
||
|
||
|
||
def test_buildings_under_the_street_go_and_the_others_stay():
|
||
result = apply_street(world(), STREET)
|
||
assert [b[0].bounds for b in result.buildings] == [(20.0, 400.0, 60.0, 450.0), (84.0, 450.0, 93.0, 470.0)]
|
||
|
||
|
||
def test_alley_is_cut_and_the_straight_street_joins_both_cross_streets():
|
||
result = apply_street(world(), STREET)
|
||
assert not on_road(result, 100, 500)
|
||
assert on_road(result, 150, 500) and not on_road(result, 143, 500)
|
||
assert on_road(result, 100, 650) and on_road(result, 100, 330)
|
||
assert on_road(result, 150, 650) and on_road(result, 150, 330)
|
||
|
||
|
||
def test_alley_centerline_and_short_stubs_go_and_the_street_gets_its_own():
|
||
result = apply_street(world(), STREET)
|
||
assert all(line.distance(Point(100, 500)) > 40 for line in result.centerlines)
|
||
assert LineString([(150, 660), (150, 320)]) in result.centerlines
|
||
assert all(line.length >= 15 for line in result.centerlines)
|
||
|
||
|
||
def test_freed_ground_becomes_urban_but_not_under_the_street_buildings_or_old_urban():
|
||
result = apply_street(world(), STREET)
|
||
new = [poly for poly, kind in result.surfaces[2:] if kind == "urban"]
|
||
assert any(poly.contains(Point(100, 500)) for poly in new) # alte Gasse
|
||
assert any(poly.contains(Point(165, 615)) for poly in new) # alter Grundriss neben der Straße
|
||
assert not any(poly.contains(Point(150, 615)) for poly in new) # unter der neuen Straße
|
||
assert not any(poly.contains(Point(91, 460)) for poly in new) # stehendes Gebäude
|
||
assert not any(poly.contains(Point(92, 500)) for poly in new) # war schon Stadtfläche
|
||
|
||
|
||
def test_palms_on_the_new_street_go():
|
||
assert apply_street(world(), STREET).palms == [(60.0, 500.0), (140.0, 500.0)]
|