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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
"""
Name: v.extract test
Purpose: Tests v.extract and its flags/options.
Author: Sunveer Singh, Google Code-in 2017
Copyright: (C) 2017 by Sunveer Singh and the GRASS Development Team
Licence: This program is free software under the GNU General Public
License (>=v2). Read the file COPYING that comes with GRASS
for details.
"""
import os
from grass.gunittest.case import TestCase
from grass.gunittest.gmodules import SimpleModule
from grass.script.core import read_command
TABLE_1 = """cat|MAJORRDS_|ROAD_NAME|MULTILANE|PROPYEAR|OBJECTID|SHAPE_LEN
1|1|NC-50|no|0|1|4825.369405
2|2|NC-50|no|0|2|14392.589058
3|3|NC-98|no|0|3|3212.981242
4|4|NC-50|no|0|4|13391.907552
"""
TABLE_2 = """cat|onemap_pro|PERIMETER|GEOL250_|GEOL250_ID|GEO_NAME|SHAPE_area|SHAPE_len
1|963738.75|4083.97998|2|1|Zml|963738.608571|4083.979839
2|22189124|26628.261719|3|2|Zmf|22189123.2296|26628.261112
3|579286.875|3335.55835|4|3|Zml|579286.829631|3335.557182
4|20225526|33253|5|4|Zml|20225526.6368|33253.000508
5|450650720|181803.765625|6|5|Ybgg|450650731.029|181803.776199
"""
TABLE_3 = """cat|MAJORRDS_|ROAD_NAME|MULTILANE|PROPYEAR|OBJECTID|SHAPE_LEN
1|1|NC-50|no|0|1|4825.369405
2|2|NC-50|no|0|2|14392.589058
3|3|NC-98|no|0|3|3212.981242
4|4|NC-50|no|0|4|13391.907552
5|5|NC-98|no|0|5|7196.001495
6|6||no|0|6|10185.513951
7|7|US-1|yes|0|7|13655.438596
8|8||no|0|8|797.901095
9|9|NC-98|no|0|9|14772.176241
"""
class TestRasterreport(TestCase):
input = "roadsmajor"
output = "testoutput"
geology = "geology"
@classmethod
def setUpClass(cls):
cls.use_temp_region()
@classmethod
def tearDownClass(cls):
cls.del_temp_region()
def tearDown(cls):
cls.runModule("g.remove", flags="f", type="vector", name=cls.output)
def test_flagd(self):
"""Testing flag d"""
self.assertModule(
"v.extract", input=self.input, output=self.output, cats="1,2,3,4"
)
category = read_command("v.db.select", map=self.output, separator="pipe")
self.assertEqual(
first=TABLE_1.replace("\n", os.linesep),
second=category,
msg="Attribute table has wrong entries",
)
def test_cats2(self):
"""Testing cats=2"""
self.assertModule(
"v.extract",
input=self.geology,
output=self.output,
flags="d",
cats="1,2,3,4,5",
)
category = read_command("v.db.select", map=self.output, separator="pipe")
self.assertEqual(
first=TABLE_2.replace("\n", os.linesep),
second=category,
msg="Attribute table has wrong entries",
)
def test_flagt(self):
"""Testing Flag T"""
self.assertModule(
"v.extract", input=self.input, output=self.output, flags="t", cats=1
)
def test_flatr(self):
"""Testing flag r"""
self.assertModule(
"v.extract", input=self.geology, output=self.output, flags="r", cats=1
)
def test_where(self):
"""Testing where"""
self.assertModule(
"v.extract",
input=self.input,
output=self.output,
flags="d",
where="cat < 10",
)
category = read_command("v.db.select", map=self.output, separator="pipe")
self.assertEqual(
first=TABLE_3.replace("\n", os.linesep),
second=category,
msg="Attribute table has wrong entries",
)
if __name__ == "__main__":
from grass.gunittest.main import test
test()
|
