blob: 5841196471ae7db9bc151b20b9636a6872788a5b (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
import argparse
import importlib.util
import sys
def _usage():
print("pytrack_generator_tests.py \
--module_path path/to/pytrack_generator.so \
--data_path path/to/omim/data \
--user_resource_path path/to/omim/data")
def _main():
parser = argparse.ArgumentParser()
parser.add_argument(
'--module_path',
type=str
)
parser.add_argument(
'--data_path',
type=str
)
parser.add_argument(
'--user_resources_path',
type=str,
)
args = parser.parse_args(sys.argv[1:])
if not args.module_path or not args.data_path or not args.user_resources_path:
_usage()
sys.exit(2)
spec = importlib.util.spec_from_file_location("pytrack_generator", args.module_path)
ge = importlib.util.module_from_spec(spec)
spec.loader.exec_module(ge)
params = ge.Params(args.data_path, args.user_resources_path)
generator = ge.Generator(params)
points = ge.LatLonList()
points.append(ge.LatLon(55.796993, 37.537640))
points.append(ge.LatLon(55.798087, 37.539002))
result = generator.generate(points)
assert len(result) > len(points)
try:
invalid_points = ge.LatLonList()
invalid_points.append(ge.LatLon(20, 20))
invalid_points.append(ge.LatLon(20, 20))
generator.generate(invalid_points)
except ge.RouteNotFoundException as ex:
print(ex)
return
assert False
if __name__ == '__main__':
_main()
|