test_raster_algebra.py 31 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697
  1. """
  2. (C) 2013 by the GRASS Development Team
  3. This program is free software under the GNU General Public
  4. License (>=v2). Read the file COPYING that comes with GRASS
  5. for details.
  6. :authors: Soeren Gebbert
  7. """
  8. import datetime
  9. import os
  10. import grass.temporal as tgis
  11. from grass.gunittest.case import TestCase
  12. from grass.gunittest.main import test
  13. class TestTRastAlgebra(TestCase):
  14. @classmethod
  15. def setUpClass(cls):
  16. """Initiate the temporal GIS and set the region
  17. """
  18. os.putenv("GRASS_OVERWRITE", "1")
  19. tgis.init(True) # Raise on error instead of exit(1)
  20. cls.use_temp_region()
  21. cls.runModule("g.region", n=80.0, s=0.0, e=120.0,
  22. w=0.0, t=1.0, b=0.0, res=10.0)
  23. cls.runModule("r.mapcalc", quiet=True, expression="a1 = 1")
  24. cls.runModule("r.mapcalc", quiet=True, expression="a2 = 2")
  25. cls.runModule("r.mapcalc", quiet=True, expression="a3 = 3")
  26. cls.runModule("r.mapcalc", quiet=True, expression="a4 = 4")
  27. cls.runModule("r.mapcalc", quiet=True, expression="b1 = 5")
  28. cls.runModule("r.mapcalc", quiet=True, expression="b2 = 6")
  29. cls.runModule("r.mapcalc", quiet=True, expression="c1 = 7")
  30. cls.runModule("r.mapcalc", quiet=True, expression="d1 = 8")
  31. cls.runModule("r.mapcalc", quiet=True, expression="d2 = 9")
  32. cls.runModule("r.mapcalc", quiet=True, expression="d3 = 10")
  33. cls.runModule("r.mapcalc", quiet=True, expression="singletmap = 99")
  34. cls.runModule("r.mapcalc", quiet=True, expression="singlemap = 100")
  35. tgis.open_new_stds(name="A", type="strds", temporaltype="absolute",
  36. title="A", descr="A", semantic="field")
  37. tgis.open_new_stds(name="B", type="strds", temporaltype="absolute",
  38. title="B", descr="B", semantic="field")
  39. tgis.open_new_stds(name="C", type="strds", temporaltype="absolute",
  40. title="B", descr="C", semantic="field")
  41. tgis.open_new_stds(name="D", type="strds", temporaltype="absolute",
  42. title="D", descr="D", semantic="field")
  43. tgis.register_maps_in_space_time_dataset(type="raster", name="A", maps="a1,a2,a3,a4",
  44. start="2001-01-01", increment="1 day", interval=True)
  45. tgis.register_maps_in_space_time_dataset(type="raster", name="B", maps="b1,b2",
  46. start="2001-01-01", increment="2 day", interval=True)
  47. tgis.register_maps_in_space_time_dataset(type="raster", name="C", maps="c1",
  48. start="2001-01-02", increment="2 day", interval=True)
  49. tgis.register_maps_in_space_time_dataset(type="raster", name="D", maps="d1,d2,d3",
  50. start="2001-01-03", increment="1 day", interval=True)
  51. tgis.register_maps_in_space_time_dataset(type="raster", name=None, maps="singletmap",
  52. start="2001-01-03", end="2001-01-04")
  53. def tearDown(self):
  54. self.runModule("t.remove", flags="rf", inputs="R", quiet=True)
  55. @classmethod
  56. def tearDownClass(cls):
  57. """Remove the temporary region
  58. """
  59. cls.runModule("t.remove", flags="rf", inputs="A,B,C,D", quiet=True)
  60. cls.runModule("t.unregister", maps="singletmap", quiet=True)
  61. cls.del_temp_region()
  62. def test_temporal_conditional_time_dimension_bug(self):
  63. """Testing the conditional time dimension bug, that uses the time
  64. dimension of the conditional statement instead the time dimension
  65. of the then/else statement."""
  66. self.assertModule("t.rast.algebra", expression="R = if({contains}, B == 5, A - 1, A + 1)",
  67. basename="r")
  68. D = tgis.open_old_stds("R", type="strds")
  69. self.assertEqual(D.metadata.get_number_of_maps(), 4)
  70. self.assertEqual(D.metadata.get_min_min(), 0) # 1 - 1
  71. self.assertEqual(D.metadata.get_max_max(), 5) # 4 + 1
  72. start, end = D.get_absolute_time()
  73. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  74. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  75. self.assertEqual( D.check_temporal_topology(), True)
  76. self.assertEqual(D.get_granularity(), u'1 day')
  77. def test_simple_arith_hash_1(self):
  78. """Simple arithmetic test including the hash operator"""
  79. self.assertModule("t.rast.algebra", expression='R = A + (A {#, equal,l} A)', basename="r")
  80. D = tgis.open_old_stds("R", type="strds")
  81. self.assertEqual(D.metadata.get_number_of_maps(), 4)
  82. self.assertEqual(D.metadata.get_min_min(), 2)
  83. self.assertEqual(D.metadata.get_max_max(), 5)
  84. start, end = D.get_absolute_time()
  85. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  86. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  87. def test_simple_arith_td_1(self):
  88. """Simple arithmetic test"""
  89. self.assertModule("t.rast.algebra", expression='R = A + td(A)', basename="r")
  90. D = tgis.open_old_stds("R", type="strds")
  91. self.assertEqual(D.metadata.get_number_of_maps(), 4)
  92. self.assertEqual(D.metadata.get_min_min(), 2)
  93. self.assertEqual(D.metadata.get_max_max(), 5)
  94. start, end = D.get_absolute_time()
  95. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  96. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  97. def test_simple_arith_td_2(self):
  98. """Simple arithmetic test"""
  99. self.assertModule("t.rast.algebra", expression='R = A / td(A)', basename="r")
  100. D = tgis.open_old_stds("R", type="strds")
  101. self.assertEqual(D.metadata.get_number_of_maps(), 4)
  102. self.assertEqual(D.metadata.get_min_min(), 1)
  103. self.assertEqual(D.metadata.get_max_max(), 4)
  104. start, end = D.get_absolute_time()
  105. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  106. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  107. def test_simple_arith_td_3(self):
  108. """Simple arithmetic test"""
  109. self.assertModule("t.rast.algebra", expression='R = A {+,equal} td(A)', basename="r")
  110. D = tgis.open_old_stds("R", type="strds")
  111. self.assertEqual(D.metadata.get_number_of_maps(), 4)
  112. self.assertEqual(D.metadata.get_min_min(), 2)
  113. self.assertEqual(D.metadata.get_max_max(), 5)
  114. start, end = D.get_absolute_time()
  115. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  116. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  117. def test_simple_arith_td_4(self):
  118. """Simple arithmetic test"""
  119. self.assertModule("t.rast.algebra", expression='R = A {/, equal} td(A)', basename="r")
  120. D = tgis.open_old_stds("R", type="strds")
  121. self.assertEqual(D.metadata.get_number_of_maps(), 4)
  122. self.assertEqual(D.metadata.get_min_min(), 1)
  123. self.assertEqual(D.metadata.get_max_max(), 4)
  124. start, end = D.get_absolute_time()
  125. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  126. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  127. def test_simple_arith_if_1(self):
  128. """Simple arithmetic test with if condition"""
  129. self.assertModule("t.rast.algebra", expression='R = if({equal}, start_date(A) >= "2001-01-02", A + A)', basename="r")
  130. D = tgis.open_old_stds("R", type="strds")
  131. self.assertEqual(D.metadata.get_number_of_maps(), 3)
  132. self.assertEqual(D.metadata.get_min_min(), 4)
  133. self.assertEqual(D.metadata.get_max_max(), 8)
  134. start, end = D.get_absolute_time()
  135. self.assertEqual(start, datetime.datetime(2001, 1, 2))
  136. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  137. def test_simple_arith_if_2(self):
  138. """Simple arithmetic test with if condition"""
  139. self.assertModule("t.rast.algebra", expression='R = if({equal}, A#A == 1, A - A)', basename="r")
  140. D = tgis.open_old_stds("R", type="strds")
  141. self.assertEqual(D.metadata.get_number_of_maps(), 4)
  142. self.assertEqual(D.metadata.get_min_min(), 0)
  143. self.assertEqual(D.metadata.get_max_max(), 0)
  144. start, end = D.get_absolute_time()
  145. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  146. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  147. def test_complex_arith_if_1(self):
  148. """Complex arithmetic test with if condition"""
  149. self.assertModule("t.rast.algebra", expression='R = if(start_date(A) < "2001-01-03" && A#A == 1, A{+, starts,l}C, A{+, finishes,l}C)', \
  150. basename="r")
  151. D = tgis.open_old_stds("R", type="strds")
  152. self.assertEqual(D.metadata.get_number_of_maps(), 2)
  153. self.assertEqual(D.metadata.get_min_min(), 9) # 2 + 7 a2 + c1
  154. self.assertEqual(D.metadata.get_max_max(), 10) # 3 + 7 a3 + c1
  155. start, end = D.get_absolute_time()
  156. self.assertEqual(start, datetime.datetime(2001, 1, 2))
  157. self.assertEqual(end, datetime.datetime(2001, 1, 4))
  158. def test_simple_arith_1(self):
  159. """Simple arithmetic test"""
  160. self.assertModule("t.rast.algebra", expression="R = A {*, equal} A {+, equal} A", basename="r")
  161. D = tgis.open_old_stds("R", type="strds")
  162. self.assertEqual(D.metadata.get_number_of_maps(), 4)
  163. self.assertEqual(D.metadata.get_min_min(), 2) # 1*1 + 1
  164. self.assertEqual(D.metadata.get_max_max(), 20) # 4*4 + 4
  165. start, end = D.get_absolute_time()
  166. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  167. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  168. def test_simple_arith_2(self):
  169. """Simple arithmetic test that creates an empty strds"""
  170. self.assertModule("t.rast.algebra", expression="R = A {*, during} A {+, during} A", basename="r")
  171. D = tgis.open_old_stds("R", type="strds")
  172. self.assertEqual(D.metadata.get_number_of_maps(), 0)
  173. def test_simple_arith_3(self):
  174. """Simple arithmetic test"""
  175. self.assertModule("t.rast.algebra", expression="R = A / A + A*A/A", basename="r")
  176. D = tgis.open_old_stds("R", type="strds")
  177. self.assertEqual(D.metadata.get_number_of_maps(), 4)
  178. self.assertEqual(D.metadata.get_min_min(), 2) # 1/1 + 1*1/1
  179. self.assertEqual(D.metadata.get_max_max(), 5) # 4/4 + 4*4/4
  180. start, end = D.get_absolute_time()
  181. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  182. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  183. def test_temporal_intersection_1(self):
  184. """Simple temporal intersection test"""
  185. self.assertModule("t.rast.algebra", expression="R = A {+,equal,i} B", basename="r")
  186. D = tgis.open_old_stds("R", type="strds")
  187. self.assertEqual(D.metadata.get_number_of_maps(), 0)
  188. def test_temporal_intersection_2(self):
  189. """Simple temporal intersection test"""
  190. self.assertModule("t.rast.algebra", expression="R = A {+,during,i} B", basename="r")
  191. D = tgis.open_old_stds("R", type="strds")
  192. self.assertEqual(D.metadata.get_number_of_maps(), 4)
  193. self.assertEqual(D.metadata.get_min_min(), 6) # 1 + 5
  194. self.assertEqual(D.metadata.get_max_max(), 10) # 4 + 6
  195. start, end = D.get_absolute_time()
  196. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  197. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  198. def test_temporal_intersection_3(self):
  199. """Simple temporal intersection test"""
  200. self.assertModule("t.rast.algebra", expression="R = A {+,starts,i} B", basename="r")
  201. D = tgis.open_old_stds("R", type="strds")
  202. self.assertEqual(D.metadata.get_number_of_maps(), 2)
  203. self.assertEqual(D.metadata.get_min_min(), 6) # 1 + 5
  204. self.assertEqual(D.metadata.get_max_max(), 9) # 3 + 6
  205. start, end = D.get_absolute_time()
  206. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  207. self.assertEqual(end, datetime.datetime(2001, 1, 4))
  208. def test_temporal_intersection_4(self):
  209. """Simple temporal intersection test"""
  210. self.assertModule("t.rast.algebra", expression="R = A {+,finishes,intersect} B", basename="r")
  211. D = tgis.open_old_stds("R", type="strds")
  212. self.assertEqual(D.metadata.get_number_of_maps(), 2)
  213. self.assertEqual(D.metadata.get_min_min(), 7) # 2 + 5
  214. self.assertEqual(D.metadata.get_max_max(), 10) # 4 + 6
  215. start, end = D.get_absolute_time()
  216. self.assertEqual(start, datetime.datetime(2001, 1, 2))
  217. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  218. def test_temporal_intersection_5(self):
  219. """Simple temporal intersection test"""
  220. self.assertModule("t.rast.algebra", expression="R = A {+,starts|finishes,i} B", basename="r")
  221. D = tgis.open_old_stds("R", type="strds")
  222. self.assertEqual(D.metadata.get_number_of_maps(), 4)
  223. self.assertEqual(D.metadata.get_min_min(), 6) # 1 + 5
  224. self.assertEqual(D.metadata.get_max_max(), 10) # 4 + 6
  225. start, end = D.get_absolute_time()
  226. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  227. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  228. def test_temporal_intersection_6(self):
  229. """Simple temporal intersection test"""
  230. self.assertModule("t.rast.algebra", expression="R = B {+,overlaps,u} C", basename="r")
  231. D = tgis.open_old_stds("R", type="strds")
  232. self.assertEqual(D.metadata.get_number_of_maps(), 1)
  233. self.assertEqual(D.metadata.get_min_min(), 12) # 5 + 7
  234. self.assertEqual(D.metadata.get_max_max(), 12) # 5 + 7
  235. start, end = D.get_absolute_time()
  236. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  237. self.assertEqual(end, datetime.datetime(2001, 1, 4))
  238. def test_temporal_intersection_7(self):
  239. """Simple temporal intersection test"""
  240. self.assertModule("t.rast.algebra", expression="R = B {+,overlapped,u} C", basename="r")
  241. D = tgis.open_old_stds("R", type="strds")
  242. self.assertEqual(D.metadata.get_number_of_maps(), 1)
  243. self.assertEqual(D.metadata.get_min_min(), 13) # 6 + 7
  244. self.assertEqual(D.metadata.get_max_max(), 13) # 6 + 7
  245. start, end = D.get_absolute_time()
  246. self.assertEqual(start, datetime.datetime(2001, 1, 2))
  247. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  248. def test_temporal_intersection_8(self):
  249. """Simple temporal intersection test"""
  250. self.assertModule("t.rast.algebra", expression='R = A {+,during,l} buff_t(C, "1 day") ',
  251. basename="r")
  252. D = tgis.open_old_stds("R", type="strds")
  253. self.assertEqual(D.metadata.get_number_of_maps(), 4)
  254. self.assertEqual(D.metadata.get_min_min(), 8) # 1 + 7 a1 + c1
  255. self.assertEqual(D.metadata.get_max_max(), 11) # 4 + 7 a4 + c1
  256. start, end = D.get_absolute_time()
  257. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  258. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  259. def test_temporal_neighbors_1(self):
  260. """Simple temporal neighborhood computation test"""
  261. self.assertModule("t.rast.algebra", expression='R = A[-1] + A[1]',
  262. basename="r")
  263. D = tgis.open_old_stds("R", type="strds")
  264. self.assertEqual(D.metadata.get_number_of_maps(), 2)
  265. self.assertEqual(D.metadata.get_min_min(), 4) # 1 + 3
  266. self.assertEqual(D.metadata.get_max_max(), 6) # 2 + 4
  267. start, end = D.get_absolute_time()
  268. self.assertEqual(start, datetime.datetime(2001, 1, 2))
  269. self.assertEqual(end, datetime.datetime(2001, 1, 4))
  270. def test_temporal_neighbors_2(self):
  271. """Simple temporal neighborhood computation test"""
  272. self.assertModule("t.rast.algebra", expression='R = A[0,0,-1] + A[0,0,1]',
  273. basename="r")
  274. D = tgis.open_old_stds("R", type="strds")
  275. self.assertEqual(D.metadata.get_number_of_maps(), 2)
  276. self.assertEqual(D.metadata.get_min_min(), 4) # 1 + 3
  277. self.assertEqual(D.metadata.get_max_max(), 6) # 2 + 4
  278. start, end = D.get_absolute_time()
  279. self.assertEqual(start, datetime.datetime(2001, 1, 2))
  280. self.assertEqual(end, datetime.datetime(2001, 1, 4))
  281. def test_tmap_function1(self):
  282. """Testing the tmap function. """
  283. self.assertModule("t.rast.algebra", expression='R = tmap(singletmap)', basename="r")
  284. D = tgis.open_old_stds("R", type="strds")
  285. maplist = D.get_registered_maps_as_objects()
  286. self.assertEqual(D.metadata.get_number_of_maps(), 1)
  287. self.assertEqual(D.metadata.get_min_min(), 99)
  288. self.assertEqual(D.metadata.get_max_max(), 99)
  289. start, end = D.get_absolute_time()
  290. self.assertEqual(start, datetime.datetime(2001, 1, 3))
  291. self.assertEqual(end, datetime.datetime(2001, 1, 4))
  292. self.assertEqual( D.check_temporal_topology(), True)
  293. self.assertEqual(D.get_granularity(), u'1 day')
  294. def test_tmap_function2(self):
  295. """Testing the tmap function. """
  296. self.assertModule("t.rast.algebra", expression='R = tmap(singletmap) + 1', basename="r")
  297. D = tgis.open_old_stds("R", type="strds")
  298. maplist = D.get_registered_maps_as_objects()
  299. self.assertEqual(D.metadata.get_number_of_maps(), 1)
  300. self.assertEqual(D.metadata.get_min_min(), 100)
  301. self.assertEqual(D.metadata.get_max_max(), 100)
  302. start, end = D.get_absolute_time()
  303. self.assertEqual(start, datetime.datetime(2001, 1, 3))
  304. self.assertEqual(end, datetime.datetime(2001, 1, 4))
  305. self.assertEqual( D.check_temporal_topology(), True)
  306. self.assertEqual(D.get_granularity(), u'1 day')
  307. def test_map_function1(self):
  308. """Testing the map function. """
  309. self.assertModule("t.rast.algebra", expression='R = map(singlemap) + A', basename="r")
  310. D = tgis.open_old_stds("R", type="strds")
  311. maplist = D.get_registered_maps_as_objects()
  312. self.assertEqual(D.metadata.get_number_of_maps(), 4)
  313. self.assertEqual(D.metadata.get_min_min(), 101)
  314. self.assertEqual(D.metadata.get_max_max(), 104)
  315. start, end = D.get_absolute_time()
  316. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  317. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  318. self.assertEqual( D.check_temporal_topology(), True)
  319. self.assertEqual(D.get_granularity(), u'1 day')
  320. def test_map_function2(self):
  321. """Testing the map function. """
  322. self.assertModule("t.rast.algebra", expression='R = A * map(singlemap)', basename="r")
  323. D = tgis.open_old_stds("R", type="strds")
  324. maplist = D.get_registered_maps_as_objects()
  325. self.assertEqual(D.metadata.get_number_of_maps(), 4)
  326. self.assertEqual(D.metadata.get_min_min(), 100)
  327. self.assertEqual(D.metadata.get_max_max(), 400)
  328. start, end = D.get_absolute_time()
  329. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  330. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  331. self.assertEqual( D.check_temporal_topology(), True)
  332. self.assertEqual(D.get_granularity(), u'1 day')
  333. def test_temporal_select(self):
  334. """Testing the temporal select operator. """
  335. self.assertModule("t.rast.algebra", expression="R = A : A", basename="r")
  336. D = tgis.open_old_stds("R", type="strds")
  337. self.assertEqual(D.metadata.get_number_of_maps(), 4)
  338. self.assertEqual(D.metadata.get_min_min(), 1)
  339. self.assertEqual(D.metadata.get_max_max(), 4)
  340. start, end = D.get_absolute_time()
  341. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  342. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  343. self.assertEqual( D.check_temporal_topology(), True)
  344. self.assertEqual(D.get_granularity(), u'1 day')
  345. def test_temporal_select(self):
  346. """Testing the temporal select operator. """
  347. self.assertModule("t.rast.algebra", expression="R = A : D", basename="r")
  348. D = tgis.open_old_stds("R", type="strds")
  349. self.assertEqual(D.metadata.get_number_of_maps(), 2)
  350. self.assertEqual(D.metadata.get_min_min(), 3)
  351. self.assertEqual(D.metadata.get_max_max(), 4)
  352. start, end = D.get_absolute_time()
  353. self.assertEqual(start, datetime.datetime(2001, 1, 3))
  354. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  355. self.assertEqual( D.check_temporal_topology(), True)
  356. self.assertEqual(D.get_granularity(), u'1 day')
  357. def test_temporal_select_operators1(self):
  358. """Testing the temporal select operator. Including temporal relations. """
  359. self.assertModule("t.rast.algebra", expression="R = A : D", basename="r")
  360. D = tgis.open_old_stds("R", type="strds")
  361. self.assertEqual(D.metadata.get_number_of_maps(), 2)
  362. self.assertEqual(D.metadata.get_min_min(), 3)
  363. self.assertEqual(D.metadata.get_max_max(), 4)
  364. start, end = D.get_absolute_time()
  365. self.assertEqual(start, datetime.datetime(2001, 1, 3))
  366. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  367. self.assertEqual( D.check_temporal_topology(), True)
  368. self.assertEqual(D.get_granularity(), u'1 day')
  369. def test_temporal_select_operators2(self):
  370. """Testing the temporal select operator. Including temporal relations. """
  371. self.assertModule("t.rast.algebra", expression="R = A {!:,during} C", basename="r")
  372. D = tgis.open_old_stds("R", type="strds")
  373. self.assertEqual(D.metadata.get_number_of_maps(), 2)
  374. self.assertEqual(D.metadata.get_min_min(), 1)
  375. self.assertEqual(D.metadata.get_max_max(), 4)
  376. start, end = D.get_absolute_time()
  377. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  378. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  379. self.assertEqual( D.check_temporal_topology(), True)
  380. self.assertEqual(D.get_granularity(), u'1 day')
  381. def test_temporal_select_operators3(self):
  382. """Testing the temporal select operator. Including temporal relations and
  383. different temporal operators (lr|+&)"""
  384. self.assertModule("t.rast.algebra", expression="R = A {:,during,d} B", basename="r")
  385. D = tgis.open_old_stds("R", type="strds")
  386. self.assertEqual(D.metadata.get_number_of_maps(), 4)
  387. self.assertEqual(D.metadata.get_min_min(), 1)
  388. self.assertEqual(D.metadata.get_max_max(), 4)
  389. start, end = D.get_absolute_time()
  390. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  391. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  392. self.assertEqual( D.check_temporal_topology(), False)
  393. self.assertEqual(D.get_granularity(), u'2 days')
  394. def test_temporal_select_operators4(self):
  395. """Testing the temporal select operator. Including temporal relations and
  396. different temporal operators (lr|+&)"""
  397. self.assertModule("t.rast.algebra", expression="R = A {:,equal|during,r} C", basename="r")
  398. D = tgis.open_old_stds("R", type="strds")
  399. maplist = D.get_registered_maps_as_objects()
  400. for map_i in maplist:
  401. start_map, end_map = map_i.get_absolute_time()
  402. self.assertEqual(start_map, datetime.datetime(2001, 1, 2))
  403. self.assertEqual(end_map, datetime.datetime(2001, 1, 4))
  404. self.assertEqual(D.metadata.get_number_of_maps(), 2)
  405. self.assertEqual(D.metadata.get_min_min(), 2)
  406. self.assertEqual(D.metadata.get_max_max(), 3)
  407. start, end = D.get_absolute_time()
  408. self.assertEqual(start, datetime.datetime(2001, 1, 2))
  409. self.assertEqual(end, datetime.datetime(2001, 1, 4))
  410. self.assertEqual( D.check_temporal_topology(), False)
  411. self.assertEqual(D.get_granularity(), u'2 days')
  412. def test_temporal_hash_operator1(self):
  413. """Testing the temporal hash operator in the raster algebra. """
  414. self.assertModule("t.rast.algebra", expression="R = if(A # D == 1, A)", basename="r")
  415. D = tgis.open_old_stds("R", type="strds")
  416. self.assertEqual(D.metadata.get_number_of_maps(), 2)
  417. self.assertEqual(D.metadata.get_min_min(), 3)
  418. self.assertEqual(D.metadata.get_max_max(), 4)
  419. start, end = D.get_absolute_time()
  420. self.assertEqual(start, datetime.datetime(2001, 1, 3))
  421. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  422. self.assertEqual( D.check_temporal_topology(), True)
  423. self.assertEqual(D.get_granularity(), u'1 day')
  424. def test_temporal_hash_operator2(self):
  425. """Testing the temporal hash operator in the raster algebra. """
  426. self.assertModule("t.rast.algebra", expression="R = A # D", basename="r")
  427. D = tgis.open_old_stds("R", type="strds")
  428. self.assertEqual(D.metadata.get_number_of_maps(), 2)
  429. self.assertEqual(D.metadata.get_min_min(), 1)
  430. self.assertEqual(D.metadata.get_max_max(), 1)
  431. start, end = D.get_absolute_time()
  432. self.assertEqual(start, datetime.datetime(2001, 1, 3))
  433. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  434. self.assertEqual( D.check_temporal_topology(), True)
  435. self.assertEqual(D.get_granularity(), u'1 day')
  436. def test_temporal_hash_operator3(self):
  437. """Testing the temporal hash operator in the raster algebra. """
  438. self.assertModule("t.rast.algebra", expression="R = C {#,contains} A", basename="r")
  439. D = tgis.open_old_stds("R", type="strds")
  440. self.assertEqual(D.metadata.get_number_of_maps(), 1)
  441. self.assertEqual(D.metadata.get_min_min(), 2)
  442. self.assertEqual(D.metadata.get_max_max(), 2)
  443. start, end = D.get_absolute_time()
  444. self.assertEqual(start, datetime.datetime(2001, 1, 2))
  445. self.assertEqual(end, datetime.datetime(2001, 1, 4))
  446. self.assertEqual( D.check_temporal_topology(), True)
  447. self.assertEqual(D.get_granularity(), u'2 days')
  448. def test_temporal_hash_operator4(self):
  449. """Testing the temporal hash operator in the raster algebra. """
  450. self.assertModule("t.rast.algebra", expression="R = if({contains},A # D == 1, C {#,contains} A)", basename="r")
  451. D = tgis.open_old_stds("R", type="strds")
  452. self.assertEqual(D.metadata.get_number_of_maps(), 1)
  453. self.assertEqual(D.metadata.get_min_min(), 2)
  454. self.assertEqual(D.metadata.get_max_max(), 2)
  455. start, end = D.get_absolute_time()
  456. self.assertEqual(start, datetime.datetime(2001, 1, 2))
  457. self.assertEqual(end, datetime.datetime(2001, 1, 4))
  458. self.assertEqual( D.check_temporal_topology(), True)
  459. self.assertEqual(D.get_granularity(), u'2 days')
  460. def test_raster_arithmetic_relation_1(self):
  461. """Arithmetic test with temporal intersection"""
  462. self.assertModule("t.rast.algebra", expression="R = B {+,contains,l} A ", basename="r")
  463. D = tgis.open_old_stds("R", type="strds")
  464. self.assertEqual(D.metadata.get_number_of_maps(), 2)
  465. self.assertEqual(D.metadata.get_min_min(), 8)
  466. self.assertEqual(D.metadata.get_max_max(), 13)
  467. start, end = D.get_absolute_time()
  468. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  469. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  470. self.assertEqual( D.check_temporal_topology(), True)
  471. self.assertEqual(D.get_granularity(), u'2 days')
  472. def test_raster_arithmetic_relation_2(self):
  473. """Arithmetic test with temporal intersection"""
  474. self.assertModule("t.rast.algebra", expression="R = B {*,contains,l} A ", basename="r")
  475. D = tgis.open_old_stds("R", type="strds")
  476. self.assertEqual(D.metadata.get_number_of_maps(), 2)
  477. self.assertEqual(D.metadata.get_min_min(), 10)
  478. self.assertEqual(D.metadata.get_max_max(), 72)
  479. start, end = D.get_absolute_time()
  480. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  481. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  482. self.assertEqual( D.check_temporal_topology(), True)
  483. self.assertEqual(D.get_granularity(), u'2 days')
  484. def test_raster_arithmetic_relation_3(self):
  485. """Arithmetic test with temporal intersection"""
  486. self.assertModule("t.rast.algebra", expression="R = B {+,contains,l} A ", basename="r")
  487. D = tgis.open_old_stds("R", type="strds")
  488. self.assertEqual(D.metadata.get_number_of_maps(), 2)
  489. self.assertEqual(D.metadata.get_min_min(), 8)
  490. self.assertEqual(D.metadata.get_max_max(), 13)
  491. start, end = D.get_absolute_time()
  492. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  493. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  494. self.assertEqual( D.check_temporal_topology(), True)
  495. self.assertEqual(D.get_granularity(), u'2 days')
  496. def test_raster_arithmetic_relation_4(self):
  497. """Arithmetic test with temporal intersection"""
  498. self.assertModule("t.rast.algebra", expression="R = B {+,contains,r} A ", basename="r")
  499. D = tgis.open_old_stds("R", type="strds")
  500. self.assertEqual(D.metadata.get_number_of_maps(),4)
  501. self.assertEqual(D.metadata.get_min_min(), 8)
  502. self.assertEqual(D.metadata.get_max_max(), 13)
  503. start, end = D.get_absolute_time()
  504. self.assertEqual(start, datetime.datetime(2001, 1, 1))
  505. self.assertEqual(end, datetime.datetime(2001, 1, 5))
  506. self.assertEqual( D.check_temporal_topology(), True)
  507. self.assertEqual(D.get_granularity(), u'1 day')
  508. def test_raster_arithmetic_relation_5(self):
  509. """Complex arithmetic test with temporal intersection"""
  510. self.assertModule("t.rast.algebra", expression="R = tmap(singletmap) {+,equal| precedes| follows,l} A + map(singlemap)", basename="r")
  511. D = tgis.open_old_stds("R", type="strds")
  512. self.assertEqual(D.metadata.get_number_of_maps(),1)
  513. self.assertEqual(D.metadata.get_min_min(), 208)
  514. self.assertEqual(D.metadata.get_max_max(), 208)
  515. start, end = D.get_absolute_time()
  516. self.assertEqual(start, datetime.datetime(2001, 1, 3))
  517. self.assertEqual(end, datetime.datetime(2001, 1, 4))
  518. self.assertEqual( D.check_temporal_topology(), True)
  519. self.assertEqual(D.get_granularity(), u'1 day')
  520. if __name__ == '__main__':
  521. test()