mirror of
https://github.com/NixOS/nixpkgs.git
synced 2024-11-26 00:43:20 +00:00
31 lines
724 B
Diff
31 lines
724 B
Diff
diff --git a/test/test.py b/test/test.py
|
|
index e02e259..1452fda 100644
|
|
--- a/test/test.py
|
|
+++ b/test/test.py
|
|
@@ -1,13 +1,24 @@
|
|
import os, shutil
|
|
import sys
|
|
import json
|
|
-from nose import with_setup
|
|
from mbutil import mbtiles_to_disk, disk_to_mbtiles
|
|
|
|
def clear_data():
|
|
try: shutil.rmtree('test/output')
|
|
except Exception: pass
|
|
|
|
+
|
|
+def with_setup(setup_func, teardown_func):
|
|
+ def wrapper(func):
|
|
+ def wrapped(*args, **kwargs):
|
|
+ setup_func()
|
|
+ func(*args, **kwargs)
|
|
+ teardown_func()
|
|
+
|
|
+ return wrapped
|
|
+ return wrapper
|
|
+
|
|
+
|
|
@with_setup(clear_data, clear_data)
|
|
def test_mbtiles_to_disk():
|
|
mbtiles_to_disk('test/data/one_tile.mbtiles', 'test/output')
|