mirror of
https://github.com/marceloprates/prettymaps.git
synced 2026-06-10 15:37:40 +02:00
Reorganized repo structure, added github workflows
This commit is contained in:
Marcelo Prates
+300
@@ -0,0 +1,300 @@
|
||||
import pytest
|
||||
import prettymaps
|
||||
import osmnx as ox
|
||||
import geopandas as gpd
|
||||
|
||||
# Helper to get a GeoDataFrame for a given query
|
||||
def get_gdf(query = "Bom Fim, Porto Alegre, Brasil"):
|
||||
return ox.geocode_to_gdf(query)
|
||||
|
||||
# Test that presets() returns a DataFrame with known presets
|
||||
def test_presets():
|
||||
df = prettymaps.presets()
|
||||
assert not df.empty
|
||||
assert 'preset' in df.columns
|
||||
assert 'params' in df.columns
|
||||
assert 'default' in df['preset'].values
|
||||
|
||||
# Test that preset() returns a Preset object with params
|
||||
def test_preset():
|
||||
p = prettymaps.preset('default')
|
||||
assert hasattr(p, 'params')
|
||||
assert isinstance(p.params, dict)
|
||||
assert 'layers' in p.params
|
||||
assert 'style' in p.params
|
||||
|
||||
# Test create_preset and reading it back
|
||||
def test_create_and_read_preset(tmp_path):
|
||||
name = 'pytest-temp-preset'
|
||||
layers = {'building': {'tags': {'building': True}}}
|
||||
style = {'building': {'fc': '#fff'}}
|
||||
prettymaps.create_preset(name, layers=layers, style=style)
|
||||
p = prettymaps.preset(name)
|
||||
assert p.params['layers'] == layers
|
||||
assert p.params['style'] == style
|
||||
|
||||
# Test Subplot class
|
||||
def test_subplot():
|
||||
s = prettymaps.Subplot('Porto Alegre', style={'building': {'fc': '#fff'}})
|
||||
assert s.query == 'Porto Alegre'
|
||||
assert 'style' in s.kwargs
|
||||
|
||||
# Test plot() basic call returns expected attributes (mocked heavy work)
|
||||
def test_plot_smoke(monkeypatch):
|
||||
import prettymaps.draw as draw
|
||||
monkeypatch.setattr(draw, 'get_gdfs', lambda *a, **k: {'perimeter': get_gdf(), 'building': get_gdf()})
|
||||
monkeypatch.setattr(draw, 'create_background', lambda *a, **k: (None, 0,0,0,0,0,0))
|
||||
monkeypatch.setattr(draw, 'draw_layers', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_keypoints', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_background', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_credit', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_hillshade', lambda *a, **k: None)
|
||||
result = prettymaps.plot('Porto Alegre', show=False)
|
||||
assert hasattr(result, 'geodataframes')
|
||||
assert hasattr(result, 'fig')
|
||||
assert hasattr(result, 'ax')
|
||||
|
||||
# Test multiplot() with two subplots (mocked)
|
||||
def test_multiplot_smoke(monkeypatch):
|
||||
import prettymaps.draw as draw
|
||||
monkeypatch.setattr(draw, 'plot', lambda *a, **k: type('FakePlot', (), {'geodataframes': {}, 'fig': None, 'ax': None, 'background': None, 'keypoints': None})())
|
||||
s1 = prettymaps.Subplot('Cidade Baixa, Porto Alegre')
|
||||
s2 = prettymaps.Subplot('Bom Fim, Porto Alegre')
|
||||
prettymaps.multiplot(s1, s2, figsize=(4,4))
|
||||
|
||||
# Test plot() with a (lat, lon) tuple as query (mocked)
|
||||
def test_plot_latlon_tuple(monkeypatch):
|
||||
import prettymaps.draw as draw
|
||||
monkeypatch.setattr(draw, 'get_gdfs', lambda *a, **k: {'perimeter': get_gdf(), 'building': get_gdf()})
|
||||
monkeypatch.setattr(draw, 'create_background', lambda *a, **k: (None, 0,0,0,0,0,0))
|
||||
monkeypatch.setattr(draw, 'draw_layers', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_keypoints', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_background', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_credit', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_hillshade', lambda *a, **k: None)
|
||||
result = prettymaps.plot((41.39, 2.17), show=False)
|
||||
assert hasattr(result, 'geodataframes')
|
||||
assert hasattr(result, 'fig')
|
||||
assert hasattr(result, 'ax')
|
||||
|
||||
# Test plot() with custom layers and style (mocked)
|
||||
def test_plot_custom_layers_style(monkeypatch):
|
||||
import prettymaps.draw as draw
|
||||
monkeypatch.setattr(draw, 'get_gdfs', lambda *a, **k: {'perimeter': get_gdf(), 'building': get_gdf()})
|
||||
monkeypatch.setattr(draw, 'create_background', lambda *a, **k: (None, 0,0,0,0,0,0))
|
||||
monkeypatch.setattr(draw, 'draw_layers', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_keypoints', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_background', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_credit', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_hillshade', lambda *a, **k: None)
|
||||
layers = {'building': {'tags': {'building': True}}}
|
||||
style = {'building': {'fc': '#fff'}}
|
||||
result = prettymaps.plot('Porto Alegre', layers=layers, style=style, show=False)
|
||||
assert hasattr(result, 'geodataframes')
|
||||
|
||||
# Test plot() with radius set to False (mocked)
|
||||
def test_plot_radius_false(monkeypatch):
|
||||
import prettymaps.draw as draw
|
||||
monkeypatch.setattr(draw, 'get_gdfs', lambda *a, **k: {'perimeter': get_gdf(), 'building': get_gdf()})
|
||||
monkeypatch.setattr(draw, 'create_background', lambda *a, **k: (None, 0,0,0,0,0,0))
|
||||
monkeypatch.setattr(draw, 'draw_layers', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_keypoints', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_background', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_credit', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_hillshade', lambda *a, **k: None)
|
||||
result = prettymaps.plot('Porto Alegre', radius=False, show=False)
|
||||
assert hasattr(result, 'geodataframes')
|
||||
|
||||
# Test plot() with keypoints argument (mocked)
|
||||
def test_plot_keypoints(monkeypatch):
|
||||
import prettymaps.draw as draw
|
||||
monkeypatch.setattr(draw, 'get_gdfs', lambda *a, **k: {'perimeter': get_gdf(), 'building': get_gdf()})
|
||||
monkeypatch.setattr(draw, 'create_background', lambda *a, **k: (None, 0,0,0,0,0,0))
|
||||
monkeypatch.setattr(draw, 'draw_layers', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_keypoints', lambda *a, **k: 'mocked_keypoints')
|
||||
monkeypatch.setattr(draw, 'draw_background', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_credit', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_hillshade', lambda *a, **k: None)
|
||||
keypoints = {'tags': {'natural': ['beach']}, 'specific': {'pedra branca': {'tags': {'natural': ['peak']}}}}
|
||||
result = prettymaps.plot('Garopaba', keypoints=keypoints, show=False)
|
||||
assert result.keypoints == 'mocked_keypoints'
|
||||
|
||||
# Test plot() with postprocessing function (mocked)
|
||||
def test_plot_postprocessing(monkeypatch):
|
||||
import prettymaps.draw as draw
|
||||
monkeypatch.setattr(draw, 'get_gdfs', lambda *a, **k: {'perimeter': get_gdf(), 'building': get_gdf()})
|
||||
monkeypatch.setattr(draw, 'create_background', lambda *a, **k: (None, 0,0,0,0,0,0))
|
||||
monkeypatch.setattr(draw, 'draw_layers', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_keypoints', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_background', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_credit', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_hillshade', lambda *a, **k: None)
|
||||
def postproc(gdfs):
|
||||
gdfs['custom'] = 123
|
||||
return gdfs
|
||||
result = prettymaps.plot('Porto Alegre', postprocessing=postproc, show=False)
|
||||
assert result.geodataframes['custom'] == 123
|
||||
|
||||
# Test plot() with save_as argument (mocked file save)
|
||||
def test_plot_save_as(monkeypatch, tmp_path):
|
||||
import prettymaps.draw as draw
|
||||
monkeypatch.setattr(draw, 'get_gdfs', lambda *a, **k: {'perimeter': get_gdf(), 'building': get_gdf()})
|
||||
monkeypatch.setattr(draw, 'create_background', lambda *a, **k: (None, 0,0,0,0,0,0))
|
||||
monkeypatch.setattr(draw, 'draw_layers', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_keypoints', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_background', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_credit', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_hillshade', lambda *a, **k: None)
|
||||
monkeypatch.setattr('matplotlib.pyplot.savefig', lambda *a, **k: None)
|
||||
save_path = tmp_path / 'out.png'
|
||||
result = prettymaps.plot('Porto Alegre', save_as=str(save_path), show=False)
|
||||
assert hasattr(result, 'geodataframes')
|
||||
|
||||
# Test plot() with circle, radius, and dilate arguments (mocked)
|
||||
def test_plot_circle_radius_dilate(monkeypatch):
|
||||
import prettymaps.draw as draw
|
||||
monkeypatch.setattr(draw, 'get_gdfs', lambda *a, **k: {'perimeter': get_gdf(), 'building': get_gdf()})
|
||||
monkeypatch.setattr(draw, 'create_background', lambda *a, **k: (None, 0,0,0,0,0,0))
|
||||
monkeypatch.setattr(draw, 'draw_layers', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_keypoints', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_background', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_credit', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_hillshade', lambda *a, **k: None)
|
||||
result = prettymaps.plot('Porto Alegre', circle=True, radius=1000, dilate=50, show=False)
|
||||
assert hasattr(result, 'geodataframes')
|
||||
|
||||
# Test plot() with mode='plotter' (mocked vsketch)
|
||||
def test_plot_mode_plotter(monkeypatch):
|
||||
import prettymaps.draw as draw
|
||||
class DummyVsk:
|
||||
def display(self):
|
||||
pass
|
||||
monkeypatch.setattr(draw, 'get_gdfs', lambda *a, **k: {'perimeter': get_gdf(), 'building': get_gdf()})
|
||||
monkeypatch.setattr(draw, 'create_background', lambda *a, **k: (None, 0,0,0,0,0,0))
|
||||
monkeypatch.setattr(draw, 'draw_layers', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_keypoints', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_background', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_credit', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_hillshade', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'init_plot', lambda *a, **k: (None, None, DummyVsk()))
|
||||
result = prettymaps.plot('Porto Alegre', mode='plotter', show=True)
|
||||
assert hasattr(result, 'geodataframes')
|
||||
|
||||
# Test multiplot() with custom Subplot styles (mocked)
|
||||
def test_multiplot_custom_subplots(monkeypatch):
|
||||
import prettymaps.draw as draw
|
||||
monkeypatch.setattr(draw, 'plot', lambda *a, **k: type('FakePlot', (), {'geodataframes': {}, 'fig': None, 'ax': None, 'background': None, 'keypoints': None})())
|
||||
s1 = prettymaps.Subplot('Cidade Baixa, Porto Alegre', style={'building': {'palette': ['#49392C']}})
|
||||
s2 = prettymaps.Subplot('Bom Fim, Porto Alegre', style={'building': {'palette': ['#BA2D0B']}})
|
||||
prettymaps.multiplot(s1, s2, figsize=(4,4))
|
||||
|
||||
# Test plot() returns object with all expected attributes (mocked)
|
||||
def test_plot_types(monkeypatch):
|
||||
import prettymaps.draw as draw
|
||||
monkeypatch.setattr(draw, 'get_gdfs', lambda *a, **k: {'perimeter': get_gdf(), 'building': get_gdf()})
|
||||
monkeypatch.setattr(draw, 'create_background', lambda *a, **k: (None, 0,0,0,0,0,0))
|
||||
monkeypatch.setattr(draw, 'draw_layers', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_keypoints', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_background', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_credit', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_hillshade', lambda *a, **k: None)
|
||||
result = prettymaps.plot('Porto Alegre', show=False)
|
||||
assert hasattr(result, 'geodataframes')
|
||||
assert hasattr(result, 'fig')
|
||||
assert hasattr(result, 'ax')
|
||||
assert hasattr(result, 'background')
|
||||
assert hasattr(result, 'keypoints')
|
||||
|
||||
# Test plot() with credit argument (mocked)
|
||||
def test_plot_credit(monkeypatch):
|
||||
import prettymaps.draw as draw
|
||||
monkeypatch.setattr(draw, 'get_gdfs', lambda *a, **k: {'perimeter': get_gdf(), 'building': get_gdf()})
|
||||
monkeypatch.setattr(draw, 'create_background', lambda *a, **k: (None, 0,0,0,0,0,0))
|
||||
monkeypatch.setattr(draw, 'draw_layers', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_keypoints', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_background', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_credit', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_hillshade', lambda *a, **k: None)
|
||||
credit = {'text': 'Test credit'}
|
||||
result = prettymaps.plot('Porto Alegre', credit=credit, show=False)
|
||||
assert hasattr(result, 'geodataframes')
|
||||
|
||||
# Test plot() with custom figsize argument (mocked)
|
||||
def test_plot_custom_figsize(monkeypatch):
|
||||
import prettymaps.draw as draw
|
||||
monkeypatch.setattr(draw, 'get_gdfs', lambda *a, **k: {'perimeter': get_gdf(), 'building': get_gdf()})
|
||||
monkeypatch.setattr(draw, 'create_background', lambda *a, **k: (None, 0,0,0,0,0,0))
|
||||
monkeypatch.setattr(draw, 'draw_layers', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_keypoints', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_background', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_credit', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_hillshade', lambda *a, **k: None)
|
||||
result = prettymaps.plot('Porto Alegre', figsize=(5,5), show=False)
|
||||
assert hasattr(result, 'geodataframes')
|
||||
|
||||
# Test plot() with x, y, scale_x, scale_y, rotation params (mocked)
|
||||
def test_plot_transform_params(monkeypatch):
|
||||
import prettymaps.draw as draw
|
||||
monkeypatch.setattr(draw, 'get_gdfs', lambda *a, **k: {'perimeter': get_gdf(), 'building': get_gdf()})
|
||||
monkeypatch.setattr(draw, 'create_background', lambda *a, **k: (None, 0,0,0,0,0,0))
|
||||
monkeypatch.setattr(draw, 'draw_layers', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_keypoints', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_background', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_credit', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_hillshade', lambda *a, **k: None)
|
||||
result = prettymaps.plot('Porto Alegre', x=1, y=2, scale_x=0.5, scale_y=0.5, rotation=45, show=False)
|
||||
assert hasattr(result, 'geodataframes')
|
||||
|
||||
# Test plot() with show=True and show=False (mocked plt.show)
|
||||
def test_plot_show_true_false(monkeypatch):
|
||||
import prettymaps.draw as draw
|
||||
monkeypatch.setattr(draw, 'get_gdfs', lambda *a, **k: {'perimeter': get_gdf(), 'building': get_gdf()})
|
||||
monkeypatch.setattr(draw, 'create_background', lambda *a, **k: (None, 0,0,0,0,0,0))
|
||||
monkeypatch.setattr(draw, 'draw_layers', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_keypoints', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_background', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_credit', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_hillshade', lambda *a, **k: None)
|
||||
monkeypatch.setattr('matplotlib.pyplot.show', lambda *a, **k: None)
|
||||
prettymaps.plot('Porto Alegre', show=True)
|
||||
prettymaps.plot('Porto Alegre', show=False)
|
||||
|
||||
# Test plot() with update_preset argument (mocked)
|
||||
def test_plot_update_preset(monkeypatch):
|
||||
import prettymaps.draw as draw
|
||||
monkeypatch.setattr(draw, 'get_gdfs', lambda *a, **k: {'perimeter': get_gdf(), 'building': get_gdf()})
|
||||
monkeypatch.setattr(draw, 'create_background', lambda *a, **k: (None, 0,0,0,0,0,0))
|
||||
monkeypatch.setattr(draw, 'draw_layers', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_keypoints', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_background', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_credit', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_hillshade', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'manage_presets', lambda *a, **k: ({'building': {}}, {}, None, None, None))
|
||||
result = prettymaps.plot('Porto Alegre', update_preset='default', show=False)
|
||||
assert hasattr(result, 'geodataframes')
|
||||
|
||||
# Test plot() with semantic=True (mocked)
|
||||
def test_plot_semantic(monkeypatch):
|
||||
import prettymaps.draw as draw
|
||||
monkeypatch.setattr(draw, 'get_gdfs', lambda *a, **k: {'perimeter': get_gdf(), 'building': get_gdf()})
|
||||
monkeypatch.setattr(draw, 'create_background', lambda *a, **k: (None, 0,0,0,0,0,0))
|
||||
monkeypatch.setattr(draw, 'draw_layers', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_keypoints', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_background', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_credit', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_hillshade', lambda *a, **k: None)
|
||||
result = prettymaps.plot('Porto Alegre', semantic=True, show=False)
|
||||
assert hasattr(result, 'geodataframes')
|
||||
|
||||
# Test plot() with adjust_aspect_ratio=False (mocked)
|
||||
def test_plot_adjust_aspect_ratio(monkeypatch):
|
||||
import prettymaps.draw as draw
|
||||
monkeypatch.setattr(draw, 'get_gdfs', lambda *a, **k: {'perimeter': get_gdf(), 'building': get_gdf()})
|
||||
monkeypatch.setattr(draw, 'create_background', lambda *a, **k: (None, 0,0,0,0,0,0))
|
||||
monkeypatch.setattr(draw, 'draw_layers', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_keypoints', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_background', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_credit', lambda *a, **k: None)
|
||||
monkeypatch.setattr(draw, 'draw_hillshade', lambda *a, **k: None)
|
||||
result = prettymaps.plot('Porto Alegre', adjust_aspect_ratio=False, show=False)
|
||||
assert hasattr(result, 'geodataframes')
|
||||
@@ -0,0 +1,23 @@
|
||||
import subprocess
|
||||
import sys
|
||||
import time
|
||||
import os
|
||||
|
||||
def test_streamlit_app_runs():
|
||||
# Start the Streamlit app in headless mode
|
||||
proc = subprocess.Popen(
|
||||
[sys.executable, "-m", "streamlit", "run", os.path.abspath("app.py"), "--server.headless", "true"],
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.PIPE,
|
||||
)
|
||||
try:
|
||||
# Wait a few seconds to see if it crashes
|
||||
time.sleep(10)
|
||||
# Check if the process is still running (should be)
|
||||
assert proc.poll() is None, "Streamlit app crashed on startup"
|
||||
finally:
|
||||
proc.terminate()
|
||||
try:
|
||||
proc.wait(timeout=5)
|
||||
except subprocess.TimeoutExpired:
|
||||
proc.kill()
|
||||
Reference in New Issue
Block a user