unit_tests.py 49 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672
  1. """!@package grass.temporal
  2. @brief GRASS Python scripting module (temporal GIS functions)
  3. Temporal GIS unit tests
  4. Usage:
  5. @code
  6. import grass.temporal as tgis
  7. tgis.test_increment_datetime_by_string()
  8. ...
  9. @endcode
  10. (C) 2008-2011 by the GRASS Development Team
  11. This program is free software under the GNU General Public
  12. License (>=v2). Read the file COPYING that comes with GRASS
  13. for details.
  14. @author Soeren Gebbert
  15. """
  16. import copy
  17. from datetime import datetime, date, time, timedelta
  18. import grass.script.core as core
  19. from temporal_granularity import *
  20. from datetime_math import *
  21. from space_time_datasets import *
  22. import grass.lib.vector as vector
  23. import grass.lib.rtree as rtree
  24. import grass.lib.gis as gis
  25. from ctypes import *
  26. # Uncomment this to detect the error
  27. core.set_raise_on_error(True)
  28. ###############################################################################
  29. def test_increment_datetime_by_string():
  30. # First test
  31. print "# Test 1"
  32. dt = datetime(2001, 9, 1, 0, 0, 0)
  33. string = "60 seconds, 4 minutes, 12 hours, 10 days, 1 weeks, 5 months, 1 years"
  34. dt1 = datetime(2003, 2, 18, 12, 5, 0)
  35. dt2 = increment_datetime_by_string(dt, string)
  36. print dt
  37. print dt2
  38. delta = dt1 - dt2
  39. if delta.days != 0 or delta.seconds != 0:
  40. core.fatal("increment computation is wrong %s" % (delta))
  41. # Second test
  42. print "# Test 2"
  43. dt = datetime(2001, 11, 1, 0, 0, 0)
  44. string = "1 months"
  45. dt1 = datetime(2001, 12, 1)
  46. dt2 = increment_datetime_by_string(dt, string)
  47. print dt
  48. print dt2
  49. delta = dt1 - dt2
  50. if delta.days != 0 or delta.seconds != 0:
  51. core.fatal("increment computation is wrong %s" % (delta))
  52. # Third test
  53. print "# Test 3"
  54. dt = datetime(2001, 11, 1, 0, 0, 0)
  55. string = "13 months"
  56. dt1 = datetime(2002, 12, 1)
  57. dt2 = increment_datetime_by_string(dt, string)
  58. print dt
  59. print dt2
  60. delta = dt1 - dt2
  61. if delta.days != 0 or delta.seconds != 0:
  62. core.fatal("increment computation is wrong %s" % (delta))
  63. # 4. test
  64. print "# Test 4"
  65. dt = datetime(2001, 1, 1, 0, 0, 0)
  66. string = "72 months"
  67. dt1 = datetime(2007, 1, 1)
  68. dt2 = increment_datetime_by_string(dt, string)
  69. print dt
  70. print dt2
  71. delta = dt1 - dt2
  72. if delta.days != 0 or delta.seconds != 0:
  73. core.fatal("increment computation is wrong %s" % (delta))
  74. ###############################################################################
  75. def test_adjust_datetime_to_granularity():
  76. # First test
  77. print "Test 1"
  78. dt = datetime(2001, 8, 8, 12, 30, 30)
  79. result = adjust_datetime_to_granularity(dt, "5 seconds")
  80. correct = datetime(2001, 8, 8, 12, 30, 30)
  81. delta = correct - result
  82. if delta.days != 0 or delta.seconds != 0:
  83. core.fatal("Granularity adjustment computation is wrong %s" % (delta))
  84. # Second test
  85. print "Test 2"
  86. result = adjust_datetime_to_granularity(dt, "20 minutes")
  87. correct = datetime(2001, 8, 8, 12, 30, 00)
  88. delta = correct - result
  89. if delta.days != 0 or delta.seconds != 0:
  90. core.fatal("Granularity adjustment computation is wrong %s" % (delta))
  91. # Third test
  92. print "Test 2"
  93. result = adjust_datetime_to_granularity(dt, "20 minutes")
  94. correct = datetime(2001, 8, 8, 12, 30, 00)
  95. delta = correct - result
  96. if delta.days != 0 or delta.seconds != 0:
  97. core.fatal("Granularity adjustment computation is wrong %s" % (delta))
  98. # 4. test
  99. print "Test 4"
  100. result = adjust_datetime_to_granularity(dt, "3 hours")
  101. correct = datetime(2001, 8, 8, 12, 00, 00)
  102. delta = correct - result
  103. if delta.days != 0 or delta.seconds != 0:
  104. core.fatal("Granularity adjustment computation is wrong %s" % (delta))
  105. # 5. test
  106. print "Test 5"
  107. result = adjust_datetime_to_granularity(dt, "5 days")
  108. correct = datetime(2001, 8, 8, 00, 00, 00)
  109. delta = correct - result
  110. if delta.days != 0 or delta.seconds != 0:
  111. core.fatal("Granularity adjustment computation is wrong %s" % (delta))
  112. # 6. test
  113. print "Test 6"
  114. result = adjust_datetime_to_granularity(dt, "2 weeks")
  115. correct = datetime(2001, 8, 6, 00, 00, 00)
  116. delta = correct - result
  117. if delta.days != 0 or delta.seconds != 0:
  118. core.fatal("Granularity adjustment computation is wrong %s" % (delta))
  119. # 7. test
  120. print "Test 7"
  121. result = adjust_datetime_to_granularity(dt, "6 months")
  122. correct = datetime(2001, 8, 1, 00, 00, 00)
  123. delta = correct - result
  124. if delta.days != 0 or delta.seconds != 0:
  125. core.fatal("Granularity adjustment computation is wrong %s" % (delta))
  126. # 8. test
  127. print "Test 8"
  128. result = adjust_datetime_to_granularity(dt, "2 years")
  129. correct = datetime(2001, 1, 1, 00, 00, 00)
  130. delta = correct - result
  131. if delta.days != 0 or delta.seconds != 0:
  132. core.fatal("Granularity adjustment computation is wrong %s" % (delta))
  133. # 9. test
  134. print "Test 9"
  135. result = adjust_datetime_to_granularity(
  136. dt, "2 years, 3 months, 5 days, 3 hours, 3 minutes, 2 seconds")
  137. correct = datetime(2001, 8, 8, 12, 30, 30)
  138. delta = correct - result
  139. if delta.days != 0 or delta.seconds != 0:
  140. core.fatal("Granularity adjustment computation is wrong %s" % (delta))
  141. # 10. test
  142. print "Test 10"
  143. result = adjust_datetime_to_granularity(dt, "3 months, 5 days, 3 minutes")
  144. correct = datetime(2001, 8, 8, 12, 30, 00)
  145. delta = correct - result
  146. if delta.days != 0 or delta.seconds != 0:
  147. core.fatal("Granularity adjustment computation is wrong %s" % (delta))
  148. # 11. test
  149. print "Test 11"
  150. result = adjust_datetime_to_granularity(dt, "3 weeks, 5 days")
  151. correct = datetime(2001, 8, 8, 00, 00, 00)
  152. delta = correct - result
  153. if delta.days != 0 or delta.seconds != 0:
  154. core.fatal("Granularity adjustment computation is wrong %s" % (delta))
  155. ###############################################################################
  156. def test_compute_datetime_delta():
  157. print "Test 1"
  158. start = datetime(2001, 1, 1, 00, 00, 00)
  159. end = datetime(2001, 1, 1, 00, 00, 00)
  160. comp = compute_datetime_delta(start, end)
  161. result = comp["second"]
  162. correct = 0
  163. delta = correct - result
  164. if delta != 0:
  165. core.fatal("Compute datetime delta is wrong %s" % (delta))
  166. print "Test 2"
  167. start = datetime(2001, 1, 1, 00, 00, 14)
  168. end = datetime(2001, 1, 1, 00, 00, 44)
  169. comp = compute_datetime_delta(start, end)
  170. result = comp["second"]
  171. correct = 30
  172. delta = correct - result
  173. if delta != 0:
  174. core.fatal("Compute datetime delta is wrong %s" % (delta))
  175. print "Test 3"
  176. start = datetime(2001, 1, 1, 00, 00, 44)
  177. end = datetime(2001, 1, 1, 00, 01, 14)
  178. comp = compute_datetime_delta(start, end)
  179. result = comp["second"]
  180. correct = 30
  181. delta = correct - result
  182. if delta != 0:
  183. core.fatal("Compute datetime delta is wrong %s" % (delta))
  184. print "Test 4"
  185. start = datetime(2001, 1, 1, 00, 00, 30)
  186. end = datetime(2001, 1, 1, 00, 05, 30)
  187. comp = compute_datetime_delta(start, end)
  188. result = comp["second"]
  189. correct = 300
  190. delta = correct - result
  191. if delta != 0:
  192. core.fatal("Compute datetime delta is wrong %s" % (delta))
  193. print "Test 5"
  194. start = datetime(2001, 1, 1, 00, 00, 00)
  195. end = datetime(2001, 1, 1, 00, 01, 00)
  196. comp = compute_datetime_delta(start, end)
  197. result = comp["minute"]
  198. correct = 1
  199. delta = correct - result
  200. if delta != 0:
  201. core.fatal("Compute datetime delta is wrong %s" % (delta))
  202. print "Test 6"
  203. start = datetime(2011, 10, 31, 00, 45, 00)
  204. end = datetime(2011, 10, 31, 01, 45, 00)
  205. comp = compute_datetime_delta(start, end)
  206. result = comp["minute"]
  207. correct = 60
  208. delta = correct - result
  209. if delta != 0:
  210. core.fatal("Compute datetime delta is wrong %s" % (delta))
  211. print "Test 7"
  212. start = datetime(2011, 10, 31, 00, 45, 00)
  213. end = datetime(2011, 10, 31, 01, 15, 00)
  214. comp = compute_datetime_delta(start, end)
  215. result = comp["minute"]
  216. correct = 30
  217. delta = correct - result
  218. if delta != 0:
  219. core.fatal("Compute datetime delta is wrong %s" % (delta))
  220. print "Test 8"
  221. start = datetime(2011, 10, 31, 00, 45, 00)
  222. end = datetime(2011, 10, 31, 12, 15, 00)
  223. comp = compute_datetime_delta(start, end)
  224. result = comp["minute"]
  225. correct = 690
  226. delta = correct - result
  227. if delta != 0:
  228. core.fatal("Compute datetime delta is wrong %s" % (delta))
  229. print "Test 9"
  230. start = datetime(2011, 10, 31, 00, 00, 00)
  231. end = datetime(2011, 10, 31, 01, 00, 00)
  232. comp = compute_datetime_delta(start, end)
  233. result = comp["hour"]
  234. correct = 1
  235. delta = correct - result
  236. if delta != 0:
  237. core.fatal("Compute datetime delta is wrong %s" % (delta))
  238. print "Test 10"
  239. start = datetime(2011, 10, 31, 00, 00, 00)
  240. end = datetime(2011, 11, 01, 01, 00, 00)
  241. comp = compute_datetime_delta(start, end)
  242. result = comp["hour"]
  243. correct = 25
  244. delta = correct - result
  245. if delta != 0:
  246. core.fatal("Compute datetime delta is wrong %s" % (delta))
  247. print "Test 11"
  248. start = datetime(2011, 10, 31, 12, 00, 00)
  249. end = datetime(2011, 11, 01, 06, 00, 00)
  250. comp = compute_datetime_delta(start, end)
  251. result = comp["hour"]
  252. correct = 18
  253. delta = correct - result
  254. if delta != 0:
  255. core.fatal("Compute datetime delta is wrong %s" % (delta))
  256. print "Test 12"
  257. start = datetime(2011, 11, 01, 00, 00, 00)
  258. end = datetime(2011, 12, 01, 01, 00, 00)
  259. comp = compute_datetime_delta(start, end)
  260. result = comp["hour"]
  261. correct = 30 * 24 + 1
  262. delta = correct - result
  263. if delta != 0:
  264. core.fatal("Compute datetime delta is wrong %s" % (delta))
  265. print "Test 13"
  266. start = datetime(2011, 11, 01, 00, 00, 00)
  267. end = datetime(2011, 11, 05, 00, 00, 00)
  268. comp = compute_datetime_delta(start, end)
  269. result = comp["day"]
  270. correct = 4
  271. delta = correct - result
  272. if delta != 0:
  273. core.fatal("Compute datetime delta is wrong %s" % (delta))
  274. print "Test 14"
  275. start = datetime(2011, 10, 06, 00, 00, 00)
  276. end = datetime(2011, 11, 05, 00, 00, 00)
  277. comp = compute_datetime_delta(start, end)
  278. result = comp["day"]
  279. correct = 30
  280. delta = correct - result
  281. if delta != 0:
  282. core.fatal("Compute datetime delta is wrong %s" % (delta))
  283. print "Test 15"
  284. start = datetime(2011, 12, 02, 00, 00, 00)
  285. end = datetime(2012, 01, 01, 00, 00, 00)
  286. comp = compute_datetime_delta(start, end)
  287. result = comp["day"]
  288. correct = 30
  289. delta = correct - result
  290. if delta != 0:
  291. core.fatal("Compute datetime delta is wrong %s" % (delta))
  292. print "Test 16"
  293. start = datetime(2011, 01, 01, 00, 00, 00)
  294. end = datetime(2011, 02, 01, 00, 00, 00)
  295. comp = compute_datetime_delta(start, end)
  296. result = comp["month"]
  297. correct = 1
  298. delta = correct - result
  299. if delta != 0:
  300. core.fatal("Compute datetime delta is wrong %s" % (delta))
  301. print "Test 17"
  302. start = datetime(2011, 12, 01, 00, 00, 00)
  303. end = datetime(2012, 01, 01, 00, 00, 00)
  304. comp = compute_datetime_delta(start, end)
  305. result = comp["month"]
  306. correct = 1
  307. delta = correct - result
  308. if delta != 0:
  309. core.fatal("Compute datetime delta is wrong %s" % (delta))
  310. print "Test 18"
  311. start = datetime(2011, 12, 01, 00, 00, 00)
  312. end = datetime(2012, 06, 01, 00, 00, 00)
  313. comp = compute_datetime_delta(start, end)
  314. result = comp["month"]
  315. correct = 6
  316. delta = correct - result
  317. if delta != 0:
  318. core.fatal("Compute datetime delta is wrong %s" % (delta))
  319. print "Test 19"
  320. start = datetime(2011, 06, 01, 00, 00, 00)
  321. end = datetime(2021, 06, 01, 00, 00, 00)
  322. comp = compute_datetime_delta(start, end)
  323. result = comp["year"]
  324. correct = 10
  325. delta = correct - result
  326. if delta != 0:
  327. core.fatal("Compute datetime delta is wrong %s" % (delta))
  328. print "Test 20"
  329. start = datetime(2011, 06, 01, 00, 00, 00)
  330. end = datetime(2012, 06, 01, 12, 00, 00)
  331. comp = compute_datetime_delta(start, end)
  332. result = comp["hour"]
  333. d = end - start
  334. correct = 12 + d.days * 24
  335. delta = correct - result
  336. if delta != 0:
  337. core.fatal("Compute datetime delta is wrong %s" % (delta))
  338. print "Test 21"
  339. start = datetime(2011, 06, 01, 00, 00, 00)
  340. end = datetime(2012, 06, 01, 12, 30, 00)
  341. comp = compute_datetime_delta(start, end)
  342. result = comp["minute"]
  343. d = end - start
  344. correct = d.days * 24 * 60 + 12 * 60 + 30
  345. delta = correct - result
  346. if delta != 0:
  347. core.fatal("Compute datetime delta is wrong %s" % (delta))
  348. print "Test 22"
  349. start = datetime(2011, 06, 01, 00, 00, 00)
  350. end = datetime(2012, 06, 01, 12, 00, 05)
  351. comp = compute_datetime_delta(start, end)
  352. result = comp["second"]
  353. d = end - start
  354. correct = 5 + 60 * 60 * 12 + d.days * 24 * 60 * 60
  355. delta = correct - result
  356. if delta != 0:
  357. core.fatal("Compute datetime delta is wrong %s" % (delta))
  358. print "Test 23"
  359. start = datetime(2011, 06, 01, 00, 00, 00)
  360. end = datetime(2012, 06, 01, 00, 30, 00)
  361. comp = compute_datetime_delta(start, end)
  362. result = comp["minute"]
  363. d = end - start
  364. correct = 30 + d.days * 24 * 60
  365. delta = correct - result
  366. if delta != 0:
  367. core.fatal("Compute datetime delta is wrong %s" % (delta))
  368. print "Test 24"
  369. start = datetime(2011, 06, 01, 00, 00, 00)
  370. end = datetime(2012, 06, 01, 00, 00, 05)
  371. comp = compute_datetime_delta(start, end)
  372. result = comp["second"]
  373. d = end - start
  374. correct = 5 + d.days * 24 * 60 * 60
  375. delta = correct - result
  376. if delta != 0:
  377. core.fatal("Compute datetime delta is wrong %s" % (delta))
  378. def test_compute_absolute_time_granularity():
  379. # First we test intervals
  380. print "Test 1"
  381. maps = []
  382. a = datetime(2001, 1, 1)
  383. increment = "1 year"
  384. for i in range(10):
  385. start = increment_datetime_by_string(a, increment, i)
  386. end = increment_datetime_by_string(a, increment, i + 1)
  387. map = RasterDataset(None)
  388. map.set_absolute_time(start, end)
  389. maps.append(map)
  390. gran = compute_absolute_time_granularity(maps)
  391. if increment != gran:
  392. core.fatal("Wrong granularity reference %s != gran %s" % (
  393. increment, gran))
  394. print "Test 2"
  395. maps = []
  396. a = datetime(2001, 1, 1)
  397. increment = "3 years"
  398. for i in range(10):
  399. start = increment_datetime_by_string(a, increment, i)
  400. end = increment_datetime_by_string(a, increment, i + 1)
  401. map = RasterDataset(None)
  402. map.set_absolute_time(start, end)
  403. maps.append(map)
  404. gran = compute_absolute_time_granularity(maps)
  405. if increment != gran:
  406. core.fatal("Wrong granularity reference %s != gran %s" % (
  407. increment, gran))
  408. print "Test 3"
  409. maps = []
  410. a = datetime(2001, 5, 1)
  411. increment = "1 month"
  412. for i in range(20):
  413. start = increment_datetime_by_string(a, increment, i)
  414. end = increment_datetime_by_string(a, increment, i + 1)
  415. map = RasterDataset(None)
  416. map.set_absolute_time(start, end)
  417. maps.append(map)
  418. gran = compute_absolute_time_granularity(maps)
  419. if increment != gran:
  420. core.fatal("Wrong granularity reference %s != gran %s" % (
  421. increment, gran))
  422. print "Test 4"
  423. maps = []
  424. a = datetime(2001, 1, 1)
  425. increment = "3 months"
  426. for i in range(20):
  427. start = increment_datetime_by_string(a, increment, i)
  428. end = increment_datetime_by_string(a, increment, i + 1)
  429. map = RasterDataset(None)
  430. map.set_absolute_time(start, end)
  431. maps.append(map)
  432. gran = compute_absolute_time_granularity(maps)
  433. if increment != gran:
  434. core.fatal("Wrong granularity reference %s != gran %s" % (
  435. increment, gran))
  436. print "Test 3"
  437. maps = []
  438. a = datetime(2001, 1, 1)
  439. increment = "1 day"
  440. for i in range(6):
  441. start = increment_datetime_by_string(a, increment, i)
  442. end = increment_datetime_by_string(a, increment, i + 1)
  443. map = RasterDataset(None)
  444. map.set_absolute_time(start, end)
  445. maps.append(map)
  446. gran = compute_absolute_time_granularity(maps)
  447. if increment != gran:
  448. core.fatal("Wrong granularity reference %s != gran %s" % (
  449. increment, gran))
  450. print "Test 4"
  451. maps = []
  452. a = datetime(2001, 1, 14)
  453. increment = "14 days"
  454. for i in range(6):
  455. start = increment_datetime_by_string(a, increment, i)
  456. end = increment_datetime_by_string(a, increment, i + 1)
  457. map = RasterDataset(None)
  458. map.set_absolute_time(start, end)
  459. maps.append(map)
  460. gran = compute_absolute_time_granularity(maps)
  461. if increment != gran:
  462. core.fatal("Wrong granularity reference %s != gran %s" % (
  463. increment, gran))
  464. print "Test 5"
  465. maps = []
  466. a = datetime(2001, 3, 1)
  467. increment = "1 month, 4 days"
  468. for i in range(20):
  469. start = increment_datetime_by_string(a, increment, i)
  470. end = increment_datetime_by_string(a, increment, i + 1)
  471. map = RasterDataset(None)
  472. map.set_absolute_time(start, end)
  473. maps.append(map)
  474. increment = "1 day"
  475. gran = compute_absolute_time_granularity(maps)
  476. if increment != gran:
  477. core.fatal("Wrong granularity reference %s != gran %s" % (
  478. increment, gran))
  479. print "Test 6"
  480. maps = []
  481. a = datetime(2001, 2, 11)
  482. increment = "1 days, 1 hours"
  483. for i in range(20):
  484. start = increment_datetime_by_string(a, increment, i)
  485. end = increment_datetime_by_string(a, increment, i + 1)
  486. map = RasterDataset(None)
  487. map.set_absolute_time(start, end)
  488. maps.append(map)
  489. increment = "25 hours"
  490. gran = compute_absolute_time_granularity(maps)
  491. if increment != gran:
  492. core.fatal("Wrong granularity reference %s != gran %s" % (
  493. increment, gran))
  494. print "Test 7"
  495. maps = []
  496. a = datetime(2001, 6, 12)
  497. increment = "6 hours"
  498. for i in range(20):
  499. start = increment_datetime_by_string(a, increment, i)
  500. end = increment_datetime_by_string(a, increment, i + 1)
  501. map = RasterDataset(None)
  502. map.set_absolute_time(start, end)
  503. maps.append(map)
  504. gran = compute_absolute_time_granularity(maps)
  505. if increment != gran:
  506. core.fatal("Wrong granularity reference %s != gran %s" % (
  507. increment, gran))
  508. print "Test 8"
  509. maps = []
  510. a = datetime(2001, 1, 1)
  511. increment = "20 minutes"
  512. for i in range(20):
  513. start = increment_datetime_by_string(a, increment, i)
  514. end = increment_datetime_by_string(a, increment, i + 1)
  515. map = RasterDataset(None)
  516. map.set_absolute_time(start, end)
  517. maps.append(map)
  518. gran = compute_absolute_time_granularity(maps)
  519. if increment != gran:
  520. core.fatal("Wrong granularity reference %s != gran %s" % (
  521. increment, gran))
  522. print "Test 9"
  523. maps = []
  524. a = datetime(2001, 1, 1)
  525. increment = "5 hours, 25 minutes"
  526. for i in range(20):
  527. start = increment_datetime_by_string(a, increment, i)
  528. end = increment_datetime_by_string(a, increment, i + 1)
  529. map = RasterDataset(None)
  530. map.set_absolute_time(start, end)
  531. maps.append(map)
  532. increment = "325 minutes"
  533. gran = compute_absolute_time_granularity(maps)
  534. if increment != gran:
  535. core.fatal("Wrong granularity reference %s != gran %s" % (
  536. increment, gran))
  537. print "Test 10"
  538. maps = []
  539. a = datetime(2001, 1, 1)
  540. increment = "5 minutes, 30 seconds"
  541. for i in range(20):
  542. start = increment_datetime_by_string(a, increment, i)
  543. end = increment_datetime_by_string(a, increment, i + 1)
  544. map = RasterDataset(None)
  545. map.set_absolute_time(start, end)
  546. maps.append(map)
  547. increment = "330 seconds"
  548. gran = compute_absolute_time_granularity(maps)
  549. if increment != gran:
  550. core.fatal("Wrong granularity reference %s != gran %s" % (
  551. increment, gran))
  552. print "Test 11"
  553. maps = []
  554. a = datetime(2001, 12, 31)
  555. increment = "60 minutes, 30 seconds"
  556. for i in range(24):
  557. start = increment_datetime_by_string(a, increment, i)
  558. end = increment_datetime_by_string(a, increment, i + 1)
  559. map = RasterDataset(None)
  560. map.set_absolute_time(start, end)
  561. maps.append(map)
  562. increment = "3630 seconds"
  563. gran = compute_absolute_time_granularity(maps)
  564. if increment != gran:
  565. core.fatal("Wrong granularity reference %s != gran %s" % (
  566. increment, gran))
  567. print "Test 12"
  568. maps = []
  569. a = datetime(2001, 12, 31, 12, 30, 30)
  570. increment = "3600 seconds"
  571. for i in range(24):
  572. start = increment_datetime_by_string(a, increment, i)
  573. end = increment_datetime_by_string(a, increment, i + 1)
  574. print start
  575. print end
  576. map = RasterDataset(None)
  577. map.set_absolute_time(start, end)
  578. maps.append(map)
  579. gran = compute_absolute_time_granularity(maps)
  580. if increment != gran:
  581. core.fatal("Wrong granularity reference %s != gran %s" % (
  582. increment, gran))
  583. # Test absolute time points
  584. print "Test 13"
  585. maps = []
  586. a = datetime(2001, 12, 31, 12, 30, 30)
  587. increment = "3600 seconds"
  588. for i in range(24):
  589. start = increment_datetime_by_string(a, increment, i)
  590. end = None
  591. map = RasterDataset(None)
  592. map.set_absolute_time(start, end)
  593. maps.append(map)
  594. gran = compute_absolute_time_granularity(maps)
  595. if increment != gran:
  596. core.fatal("Wrong granularity reference %s != gran %s" % (
  597. increment, gran))
  598. print "Test 14"
  599. maps = []
  600. a = datetime(2001, 12, 31, 00, 00, 00)
  601. increment = "20 days"
  602. for i in range(24):
  603. start = increment_datetime_by_string(a, increment, i)
  604. end = None
  605. map = RasterDataset(None)
  606. map.set_absolute_time(start, end)
  607. maps.append(map)
  608. gran = compute_absolute_time_granularity(maps)
  609. if increment != gran:
  610. core.fatal("Wrong granularity reference %s != gran %s" % (
  611. increment, gran))
  612. print "Test 15"
  613. maps = []
  614. a = datetime(2001, 12, 01, 00, 00, 00)
  615. increment = "5 months"
  616. for i in range(24):
  617. start = increment_datetime_by_string(a, increment, i)
  618. end = None
  619. map = RasterDataset(None)
  620. map.set_absolute_time(start, end)
  621. maps.append(map)
  622. gran = compute_absolute_time_granularity(maps)
  623. if increment != gran:
  624. core.fatal("Wrong granularity reference %s != gran %s" % (
  625. increment, gran))
  626. # Test absolute time interval and points
  627. print "Test 16"
  628. maps = []
  629. a = datetime(2001, 12, 31, 12, 30, 30)
  630. increment = "3600 seconds"
  631. for i in range(24):
  632. start = increment_datetime_by_string(a, increment, i)
  633. end = increment_datetime_by_string(a, increment, i + 1)
  634. map = RasterDataset(None)
  635. map.set_absolute_time(start, end)
  636. maps.append(map)
  637. a = datetime(2002, 02, 01, 12, 30, 30)
  638. for i in range(24):
  639. start = increment_datetime_by_string(a, increment, i)
  640. end = None
  641. map = RasterDataset(None)
  642. map.set_absolute_time(start, end)
  643. maps.append(map)
  644. gran = compute_absolute_time_granularity(maps)
  645. if increment != gran:
  646. core.fatal("Wrong granularity reference %s != gran %s" % (
  647. increment, gran))
  648. print "Test 17"
  649. maps = []
  650. a = datetime(2001, 1, 1)
  651. increment = "2 days"
  652. for i in range(8):
  653. start = increment_datetime_by_string(a, increment, i)
  654. end = increment_datetime_by_string(a, increment, i + 1)
  655. map = RasterDataset(None)
  656. map.set_absolute_time(start, end)
  657. maps.append(map)
  658. a = datetime(2001, 02, 02)
  659. for i in range(8):
  660. start = increment_datetime_by_string(a, increment, i)
  661. end = None
  662. map = RasterDataset(None)
  663. map.set_absolute_time(start, end)
  664. maps.append(map)
  665. gran = compute_absolute_time_granularity(maps)
  666. if increment != gran:
  667. core.fatal("Wrong granularity reference %s != gran %s" % (
  668. increment, gran))
  669. ###############################################################################
  670. def test_spatial_extent_intersection():
  671. # Generate the extents
  672. A = SpatialExtent(
  673. north=80, south=20, east=60, west=10, bottom=-50, top=50)
  674. A.print_info()
  675. B = SpatialExtent(
  676. north=80, south=20, east=60, west=10, bottom=-50, top=50)
  677. B.print_info()
  678. C = A.intersect(B)
  679. C.print_info()
  680. if C.get_north() != B.get_north() or C.get_south() != B.get_south() or \
  681. C.get_west() != B.get_west() or C.get_east() != B.get_east() or \
  682. C.get_bottom() != B.get_bottom() or C.get_top() != B.get_top():
  683. core.fatal("Wrong intersection computation")
  684. B = SpatialExtent(
  685. north=40, south=30, east=60, west=10, bottom=-50, top=50)
  686. B.print_info()
  687. C = A.intersect(B)
  688. C.print_info()
  689. if C.get_north() != B.get_north() or C.get_south() != B.get_south() or \
  690. C.get_west() != B.get_west() or C.get_east() != B.get_east() or \
  691. C.get_bottom() != B.get_bottom() or C.get_top() != B.get_top():
  692. core.fatal("Wrong intersection computation")
  693. B = SpatialExtent(
  694. north=40, south=30, east=60, west=30, bottom=-50, top=50)
  695. B.print_info()
  696. C = A.intersect(B)
  697. C.print_info()
  698. if C.get_north() != B.get_north() or C.get_south() != B.get_south() or \
  699. C.get_west() != B.get_west() or C.get_east() != B.get_east() or \
  700. C.get_bottom() != B.get_bottom() or C.get_top() != B.get_top():
  701. core.fatal("Wrong intersection computation")
  702. B = SpatialExtent(
  703. north=40, south=30, east=60, west=30, bottom=-30, top=50)
  704. B.print_info()
  705. C = A.intersect(B)
  706. C.print_info()
  707. if C.get_north() != B.get_north() or C.get_south() != B.get_south() or \
  708. C.get_west() != B.get_west() or C.get_east() != B.get_east() or \
  709. C.get_bottom() != B.get_bottom() or C.get_top() != B.get_top():
  710. core.fatal("Wrong intersection computation")
  711. B = SpatialExtent(
  712. north=40, south=30, east=60, west=30, bottom=-30, top=30)
  713. B.print_info()
  714. C = A.intersect(B)
  715. C.print_info()
  716. if C.get_north() != B.get_north() or C.get_south() != B.get_south() or \
  717. C.get_west() != B.get_west() or C.get_east() != B.get_east() or \
  718. C.get_bottom() != B.get_bottom() or C.get_top() != B.get_top():
  719. core.fatal("Wrong intersection computation")
  720. ###############################################################################
  721. def test_spatial_relations():
  722. # Generate the extents
  723. A = SpatialExtent(
  724. north=80, south=20, east=60, west=10, bottom=-50, top=50)
  725. A.print_info()
  726. B = SpatialExtent(
  727. north=80, south=20, east=60, west=10, bottom=-50, top=50)
  728. B.print_info()
  729. relation = A.spatial_relation(B)
  730. print relation
  731. if relation != "equivalent":
  732. core.fatal("Wrong spatial relation: %s" % (relation))
  733. B = SpatialExtent(
  734. north=70, south=20, east=60, west=10, bottom=-50, top=50)
  735. B.print_info()
  736. relation = A.spatial_relation_2d(B)
  737. print relation
  738. if relation != "cover":
  739. core.fatal("Wrong spatial relation: %s" % (relation))
  740. relation = A.spatial_relation(B)
  741. print relation
  742. if relation != "cover":
  743. core.fatal("Wrong spatial relation: %s" % (relation))
  744. B = SpatialExtent(
  745. north=70, south=30, east=60, west=10, bottom=-50, top=50)
  746. B.print_info()
  747. relation = A.spatial_relation_2d(B)
  748. print relation
  749. if relation != "cover":
  750. core.fatal("Wrong spatial relation: %s" % (relation))
  751. relation = A.spatial_relation(B)
  752. print relation
  753. if relation != "cover":
  754. core.fatal("Wrong spatial relation: %s" % (relation))
  755. relation = B.spatial_relation_2d(A)
  756. print relation
  757. if relation != "covered":
  758. core.fatal("Wrong spatial relation: %s" % (relation))
  759. relation = B.spatial_relation(A)
  760. print relation
  761. if relation != "covered":
  762. core.fatal("Wrong spatial relation: %s" % (relation))
  763. B = SpatialExtent(
  764. north=70, south=30, east=50, west=10, bottom=-50, top=50)
  765. B.print_info()
  766. relation = A.spatial_relation_2d(B)
  767. print relation
  768. if relation != "cover":
  769. core.fatal("Wrong spatial relation: %s" % (relation))
  770. relation = B.spatial_relation_2d(A)
  771. print relation
  772. if relation != "covered":
  773. core.fatal("Wrong spatial relation: %s" % (relation))
  774. relation = A.spatial_relation(B)
  775. print relation
  776. if relation != "cover":
  777. core.fatal("Wrong spatial relation: %s" % (relation))
  778. B = SpatialExtent(
  779. north=70, south=30, east=50, west=20, bottom=-50, top=50)
  780. relation = B.spatial_relation(A)
  781. print relation
  782. if relation != "covered":
  783. core.fatal("Wrong spatial relation: %s" % (relation))
  784. B = SpatialExtent(
  785. north=70, south=30, east=50, west=20, bottom=-50, top=50)
  786. B.print_info()
  787. relation = A.spatial_relation_2d(B)
  788. print relation
  789. if relation != "contain":
  790. core.fatal("Wrong spatial relation: %s" % (relation))
  791. relation = A.spatial_relation(B)
  792. print relation
  793. if relation != "cover":
  794. core.fatal("Wrong spatial relation: %s" % (relation))
  795. B = SpatialExtent(
  796. north=70, south=30, east=50, west=20, bottom=-40, top=50)
  797. B.print_info()
  798. relation = A.spatial_relation(B)
  799. print relation
  800. if relation != "cover":
  801. core.fatal("Wrong spatial relation: %s" % (relation))
  802. B = SpatialExtent(
  803. north=70, south=30, east=50, west=20, bottom=-40, top=40)
  804. B.print_info()
  805. relation = A.spatial_relation(B)
  806. print relation
  807. if relation != "contain":
  808. core.fatal("Wrong spatial relation: %s" % (relation))
  809. relation = B.spatial_relation(A)
  810. print relation
  811. if relation != "in":
  812. core.fatal("Wrong spatial relation: %s" % (relation))
  813. B = SpatialExtent(
  814. north=90, south=30, east=50, west=20, bottom=-40, top=40)
  815. B.print_info()
  816. relation = A.spatial_relation_2d(B)
  817. print relation
  818. if relation != "overlap":
  819. core.fatal("Wrong spatial relation: %s" % (relation))
  820. relation = A.spatial_relation(B)
  821. print relation
  822. if relation != "overlap":
  823. core.fatal("Wrong spatial relation: %s" % (relation))
  824. B = SpatialExtent(north=90, south=5, east=70, west=5, bottom=-40, top=40)
  825. A.print_info()
  826. B.print_info()
  827. relation = A.spatial_relation_2d(B)
  828. print relation
  829. if relation != "in":
  830. core.fatal("Wrong spatial relation: %s" % (relation))
  831. relation = A.spatial_relation(B)
  832. print relation
  833. if relation != "overlap":
  834. core.fatal("Wrong spatial relation: %s" % (relation))
  835. B = SpatialExtent(north=90, south=5, east=70, west=5, bottom=-40, top=60)
  836. A.print_info()
  837. B.print_info()
  838. relation = A.spatial_relation(B)
  839. print relation
  840. if relation != "overlap":
  841. core.fatal("Wrong spatial relation: %s" % (relation))
  842. B = SpatialExtent(north=90, south=5, east=70, west=5, bottom=-60, top=60)
  843. A.print_info()
  844. B.print_info()
  845. relation = A.spatial_relation(B)
  846. print relation
  847. if relation != "in":
  848. core.fatal("Wrong spatial relation: %s" % (relation))
  849. A = SpatialExtent(
  850. north=80, south=60, east=60, west=10, bottom=-50, top=50)
  851. A.print_info()
  852. B = SpatialExtent(
  853. north=60, south=20, east=60, west=10, bottom=-50, top=50)
  854. B.print_info()
  855. relation = A.spatial_relation_2d(B)
  856. print relation
  857. if relation != "meet":
  858. core.fatal("Wrong spatial relation: %s" % (relation))
  859. relation = A.spatial_relation(B)
  860. print relation
  861. if relation != "meet":
  862. core.fatal("Wrong spatial relation: %s" % (relation))
  863. A = SpatialExtent(
  864. north=60, south=40, east=60, west=10, bottom=-50, top=50)
  865. A.print_info()
  866. B = SpatialExtent(
  867. north=80, south=60, east=60, west=10, bottom=-50, top=50)
  868. B.print_info()
  869. relation = A.spatial_relation_2d(B)
  870. print relation
  871. if relation != "meet":
  872. core.fatal("Wrong spatial relation: %s" % (relation))
  873. relation = A.spatial_relation(B)
  874. print relation
  875. if relation != "meet":
  876. core.fatal("Wrong spatial relation: %s" % (relation))
  877. A = SpatialExtent(
  878. north=80, south=40, east=60, west=40, bottom=-50, top=50)
  879. A.print_info()
  880. B = SpatialExtent(
  881. north=80, south=40, east=40, west=20, bottom=-50, top=50)
  882. B.print_info()
  883. relation = A.spatial_relation_2d(B)
  884. print relation
  885. if relation != "meet":
  886. core.fatal("Wrong spatial relation: %s" % (relation))
  887. relation = A.spatial_relation(B)
  888. print relation
  889. if relation != "meet":
  890. core.fatal("Wrong spatial relation: %s" % (relation))
  891. A = SpatialExtent(
  892. north=80, south=40, east=40, west=20, bottom=-50, top=50)
  893. A.print_info()
  894. B = SpatialExtent(
  895. north=90, south=30, east=60, west=40, bottom=-50, top=50)
  896. B.print_info()
  897. relation = A.spatial_relation_2d(B)
  898. print relation
  899. if relation != "meet":
  900. core.fatal("Wrong spatial relation: %s" % (relation))
  901. relation = A.spatial_relation(B)
  902. print relation
  903. if relation != "meet":
  904. core.fatal("Wrong spatial relation: %s" % (relation))
  905. A = SpatialExtent(
  906. north=80, south=40, east=40, west=20, bottom=-50, top=50)
  907. A.print_info()
  908. B = SpatialExtent(
  909. north=70, south=50, east=60, west=40, bottom=-50, top=50)
  910. B.print_info()
  911. relation = A.spatial_relation_2d(B)
  912. print relation
  913. if relation != "meet":
  914. core.fatal("Wrong spatial relation: %s" % (relation))
  915. relation = A.spatial_relation(B)
  916. print relation
  917. if relation != "meet":
  918. core.fatal("Wrong spatial relation: %s" % (relation))
  919. A = SpatialExtent(
  920. north=80, south=40, east=40, west=20, bottom=-50, top=50)
  921. A.print_info()
  922. B = SpatialExtent(
  923. north=60, south=20, east=60, west=40, bottom=-50, top=50)
  924. B.print_info()
  925. relation = A.spatial_relation_2d(B)
  926. print relation
  927. if relation != "meet":
  928. core.fatal("Wrong spatial relation: %s" % (relation))
  929. relation = A.spatial_relation(B)
  930. print relation
  931. if relation != "meet":
  932. core.fatal("Wrong spatial relation: %s" % (relation))
  933. A = SpatialExtent(
  934. north=80, south=40, east=40, west=20, bottom=-50, top=50)
  935. A.print_info()
  936. B = SpatialExtent(
  937. north=40, south=20, east=60, west=40, bottom=-50, top=50)
  938. B.print_info()
  939. relation = A.spatial_relation_2d(B)
  940. print relation
  941. if relation != "disjoint":
  942. core.fatal("Wrong spatial relation: %s" % (relation))
  943. relation = A.spatial_relation(B)
  944. print relation
  945. if relation != "disjoint":
  946. core.fatal("Wrong spatial relation: %s" % (relation))
  947. A = SpatialExtent(
  948. north=80, south=40, east=40, west=20, bottom=-50, top=50)
  949. A.print_info()
  950. B = SpatialExtent(
  951. north=60, south=20, east=60, west=40, bottom=-60, top=60)
  952. B.print_info()
  953. relation = A.spatial_relation(B)
  954. print relation
  955. if relation != "meet":
  956. core.fatal("Wrong spatial relation: %s" % (relation))
  957. A = SpatialExtent(
  958. north=80, south=40, east=40, west=20, bottom=-50, top=50)
  959. A.print_info()
  960. B = SpatialExtent(
  961. north=90, south=30, east=60, west=40, bottom=-40, top=40)
  962. B.print_info()
  963. relation = A.spatial_relation(B)
  964. print relation
  965. if relation != "meet":
  966. core.fatal("Wrong spatial relation: %s" % (relation))
  967. A = SpatialExtent(north=80, south=40, east=60, west=20, bottom=0, top=50)
  968. A.print_info()
  969. B = SpatialExtent(north=80, south=40, east=60, west=20, bottom=-50, top=0)
  970. B.print_info()
  971. relation = A.spatial_relation(B)
  972. print relation
  973. if relation != "meet":
  974. core.fatal("Wrong spatial relation: %s" % (relation))
  975. A = SpatialExtent(north=80, south=40, east=60, west=20, bottom=0, top=50)
  976. A.print_info()
  977. B = SpatialExtent(north=80, south=50, east=60, west=30, bottom=-50, top=0)
  978. B.print_info()
  979. relation = A.spatial_relation(B)
  980. print relation
  981. if relation != "meet":
  982. core.fatal("Wrong spatial relation: %s" % (relation))
  983. A = SpatialExtent(north=80, south=40, east=60, west=20, bottom=0, top=50)
  984. A.print_info()
  985. B = SpatialExtent(north=70, south=50, east=50, west=30, bottom=-50, top=0)
  986. B.print_info()
  987. relation = A.spatial_relation(B)
  988. print relation
  989. if relation != "meet":
  990. core.fatal("Wrong spatial relation: %s" % (relation))
  991. A = SpatialExtent(north=80, south=40, east=60, west=20, bottom=0, top=50)
  992. A.print_info()
  993. B = SpatialExtent(north=90, south=30, east=70, west=10, bottom=-50, top=0)
  994. B.print_info()
  995. relation = A.spatial_relation(B)
  996. print relation
  997. if relation != "meet":
  998. core.fatal("Wrong spatial relation: %s" % (relation))
  999. A = SpatialExtent(north=80, south=40, east=60, west=20, bottom=0, top=50)
  1000. A.print_info()
  1001. B = SpatialExtent(north=70, south=30, east=50, west=10, bottom=-50, top=0)
  1002. B.print_info()
  1003. relation = A.spatial_relation(B)
  1004. print relation
  1005. if relation != "meet":
  1006. core.fatal("Wrong spatial relation: %s" % (relation))
  1007. ###
  1008. A = SpatialExtent(north=80, south=40, east=60, west=20, bottom=-50, top=0)
  1009. A.print_info()
  1010. B = SpatialExtent(north=80, south=40, east=60, west=20, bottom=0, top=50)
  1011. B.print_info()
  1012. relation = A.spatial_relation(B)
  1013. print relation
  1014. if relation != "meet":
  1015. core.fatal("Wrong spatial relation: %s" % (relation))
  1016. A = SpatialExtent(north=80, south=40, east=60, west=20, bottom=-50, top=0)
  1017. A.print_info()
  1018. B = SpatialExtent(north=80, south=50, east=60, west=30, bottom=0, top=50)
  1019. B.print_info()
  1020. relation = A.spatial_relation(B)
  1021. print relation
  1022. if relation != "meet":
  1023. core.fatal("Wrong spatial relation: %s" % (relation))
  1024. A = SpatialExtent(north=80, south=40, east=60, west=20, bottom=-50, top=0)
  1025. A.print_info()
  1026. B = SpatialExtent(north=70, south=50, east=50, west=30, bottom=0, top=50)
  1027. B.print_info()
  1028. relation = A.spatial_relation(B)
  1029. print relation
  1030. if relation != "meet":
  1031. core.fatal("Wrong spatial relation: %s" % (relation))
  1032. A = SpatialExtent(north=80, south=40, east=60, west=20, bottom=-50, top=0)
  1033. A.print_info()
  1034. B = SpatialExtent(north=90, south=30, east=70, west=10, bottom=0, top=50)
  1035. B.print_info()
  1036. relation = A.spatial_relation(B)
  1037. print relation
  1038. if relation != "meet":
  1039. core.fatal("Wrong spatial relation: %s" % (relation))
  1040. A = SpatialExtent(north=80, south=40, east=60, west=20, bottom=-50, top=0)
  1041. A.print_info()
  1042. B = SpatialExtent(north=70, south=30, east=50, west=10, bottom=0, top=50)
  1043. B.print_info()
  1044. relation = A.spatial_relation(B)
  1045. print relation
  1046. if relation != "meet":
  1047. core.fatal("Wrong spatial relation: %s" % (relation))
  1048. ###############################################################################
  1049. def test_temporal_topology_builder():
  1050. map_listA = []
  1051. _map = RasterDataset(ident="1@a")
  1052. _map.set_absolute_time(datetime(2001, 01, 01), datetime(2001, 02, 01))
  1053. map_listA.append(copy.copy(_map))
  1054. _map = RasterDataset(ident="2@a")
  1055. _map.set_absolute_time(datetime(2001, 02, 01), datetime(2001, 03, 01))
  1056. map_listA.append(copy.copy(_map))
  1057. _map = RasterDataset(ident="3@a")
  1058. _map.set_absolute_time(datetime(2001, 03, 01), datetime(2001, 04, 01))
  1059. map_listA.append(copy.copy(_map))
  1060. _map = RasterDataset(ident="4@a")
  1061. _map.set_absolute_time(datetime(2001, 04, 01), datetime(2001, 05, 01))
  1062. map_listA.append(copy.copy(_map))
  1063. _map = RasterDataset(ident="5@a")
  1064. _map.set_absolute_time(datetime(2001, 05, 01), datetime(2001, 06, 01))
  1065. map_listA.append(copy.copy(_map))
  1066. tb = SpatioTemporalTopologyBuilder()
  1067. tb.build(map_listA)
  1068. count = 0
  1069. for _map in tb:
  1070. print "[%s]" % (_map.get_name())
  1071. _map.print_topology_info()
  1072. if _map.get_id() != map_listA[count].get_id():
  1073. core.fatal("Error building temporal topology <%s> != <%s>" %
  1074. (_map.get_id(), map_listA[count].get_id()))
  1075. count += 1
  1076. map_listB = []
  1077. _map = RasterDataset(ident="1@b")
  1078. _map.set_absolute_time(datetime(2001, 01, 14), datetime(2001, 03, 14))
  1079. map_listB.append(copy.copy(_map))
  1080. _map = RasterDataset(ident="2@b")
  1081. _map.set_absolute_time(datetime(2001, 02, 01), datetime(2001, 04, 01))
  1082. map_listB.append(copy.copy(_map))
  1083. _map = RasterDataset(ident="3@b")
  1084. _map.set_absolute_time(datetime(2001, 02, 14), datetime(2001, 04, 30))
  1085. map_listB.append(copy.copy(_map))
  1086. _map = RasterDataset(ident="4@b")
  1087. _map.set_absolute_time(datetime(2001, 04, 02), datetime(2001, 04, 30))
  1088. map_listB.append(copy.copy(_map))
  1089. _map = RasterDataset(ident="5@b")
  1090. _map.set_absolute_time(datetime(2001, 05, 01), datetime(2001, 05, 14))
  1091. map_listB.append(copy.copy(_map))
  1092. tb = SpatioTemporalTopologyBuilder()
  1093. tb.build(map_listB)
  1094. # Probing some relations
  1095. if map_listB[0].get_overlapped()[0] != map_listB[1]:
  1096. core.fatal("Error building temporal topology")
  1097. if map_listB[0].get_overlapped()[1] != map_listB[2]:
  1098. core.fatal("Error building temporal topology")
  1099. if map_listB[2].get_contains()[0] != map_listB[3]:
  1100. core.fatal("Error building temporal topology")
  1101. if map_listB[3].get_during()[0] != map_listB[2]:
  1102. core.fatal("Error building temporal topology")
  1103. count = 0
  1104. for _map in tb:
  1105. print "[%s]" % (_map.get_map_id
  1106. ())
  1107. _map.print_topology_shell_info()
  1108. if _map.get_id() != map_listB[count].get_id():
  1109. core.fatal("Error building temporal topology <%s> != <%s>" %
  1110. (_map.get_id(), map_listB[count].get_id()))
  1111. count += 1
  1112. tb = SpatioTemporalTopologyBuilder()
  1113. tb.build(map_listA, map_listB)
  1114. count = 0
  1115. for _map in tb:
  1116. print "[%s]" % (_map.get_map_id())
  1117. _map.print_topology_shell_info()
  1118. if _map.get_id() != map_listA[count].get_id():
  1119. core.fatal("Error building temporal topology <%s> != <%s>" %
  1120. (_map.get_id(), map_listA[count].get_id()))
  1121. count += 1
  1122. count = 0
  1123. for _map in map_listB:
  1124. print "[%s]" % (_map.get_map_id())
  1125. _map.print_topology_shell_info()
  1126. # Probing some relations
  1127. if map_listA[3].get_follows()[0] != map_listB[1]:
  1128. core.fatal("Error building temporal topology")
  1129. if map_listA[3].get_precedes()[0] != map_listB[4]:
  1130. core.fatal("Error building temporal topology")
  1131. if map_listA[3].get_overlaps()[0] != map_listB[2]:
  1132. core.fatal("Error building temporal topology")
  1133. if map_listA[3].get_contains()[0] != map_listB[3]:
  1134. core.fatal("Error building temporal topology")
  1135. if map_listA[2].get_during()[0] != map_listB[1]:
  1136. core.fatal("Error building temporal topology")
  1137. if map_listA[2].get_during()[1] != map_listB[2]:
  1138. core.fatal("Error building temporal topology")
  1139. ###############################################################################
  1140. def test_map_list_sorting():
  1141. map_list = []
  1142. _map = RasterDataset(ident="1@a")
  1143. _map.set_absolute_time(datetime(2001, 02, 01), datetime(2001, 03, 01))
  1144. map_list.append(copy.copy(_map))
  1145. _map = RasterDataset(ident="2@a")
  1146. _map.set_absolute_time(datetime(2001, 01, 01), datetime(2001, 02, 01))
  1147. map_list.append(copy.copy(_map))
  1148. _map = RasterDataset(ident="3@a")
  1149. _map.set_absolute_time(datetime(2001, 03, 01), datetime(2001, 04, 01))
  1150. map_list.append(copy.copy(_map))
  1151. print "Original"
  1152. for _map in map_list:
  1153. print _map.get_temporal_extent_as_tuple()[0], _map.get_temporal_extent_as_tuple()[1]
  1154. print "Sorted by start time"
  1155. new_list = sorted(map_list, key=AbstractDatasetComparisonKeyStartTime)
  1156. for _map in new_list:
  1157. print _map.get_temporal_extent_as_tuple()[0], _map.get_temporal_extent_as_tuple()[1]
  1158. if new_list[0] != map_list[1]:
  1159. core.fatal("Sorting by start time failed")
  1160. if new_list[1] != map_list[0]:
  1161. core.fatal("Sorting by start time failed")
  1162. if new_list[2] != map_list[2]:
  1163. core.fatal("Sorting by start time failed")
  1164. print "Sorted by end time"
  1165. new_list = sorted(map_list, key=AbstractDatasetComparisonKeyEndTime)
  1166. for _map in new_list:
  1167. print _map.get_temporal_extent_as_tuple()[0], _map.get_temporal_extent_as_tuple()[1]
  1168. if new_list[0] != map_list[1]:
  1169. core.fatal("Sorting by end time failed")
  1170. if new_list[1] != map_list[0]:
  1171. core.fatal("Sorting by end time failed")
  1172. if new_list[2] != map_list[2]:
  1173. core.fatal("Sorting by end time failed")
  1174. ###############################################################################
  1175. def test_1d_rtree():
  1176. """Testing the rtree ctypes wrapper"""
  1177. tree = rtree.RTreeCreateTree(-1, 0, 1)
  1178. for i in xrange(10):
  1179. rect = rtree.RTreeAllocRect(tree)
  1180. rtree.RTreeSetRect1D(rect, tree, float(i - 2), float(i + 2))
  1181. rtree.RTreeInsertRect(rect, i + 1, tree)
  1182. rect = rtree.RTreeAllocRect(tree)
  1183. rtree.RTreeSetRect1D(rect, tree, 2.0, 7.0)
  1184. list_ = gis.ilist()
  1185. num = vector.RTreeSearch2(tree, rect, byref(list_))
  1186. rtree.RTreeFreeRect(rect)
  1187. # print rectangle ids
  1188. print "Number of overlapping rectangles", num
  1189. for i in xrange(list_.n_values):
  1190. print "id", list_.value[i]
  1191. rtree.RTreeDestroyTree(tree)
  1192. ###############################################################################
  1193. def test_2d_rtree():
  1194. """Testing the rtree ctypes wrapper"""
  1195. tree = rtree.RTreeCreateTree(-1, 0, 2)
  1196. for i in xrange(10):
  1197. rect = rtree.RTreeAllocRect(tree)
  1198. rtree.RTreeSetRect2D(rect, tree,
  1199. float(i - 2), float(i + 2),
  1200. float(i - 2), float(i + 2))
  1201. rtree.RTreeInsertRect(rect, i + 1, tree)
  1202. rect = rtree.RTreeAllocRect(tree)
  1203. rtree.RTreeSetRect2D(rect, tree, 2.0, 7.0, 2.0, 7.0)
  1204. list_ = gis.ilist()
  1205. num = vector.RTreeSearch2(tree, rect, byref(list_))
  1206. rtree.RTreeFreeRect(rect)
  1207. # print rectangle ids
  1208. print "Number of overlapping rectangles", num
  1209. for i in xrange(list_.n_values):
  1210. print "id", list_.value[i]
  1211. rtree.RTreeDestroyTree(tree)
  1212. ###############################################################################
  1213. def test_3d_rtree():
  1214. """Testing the rtree ctypes wrapper"""
  1215. tree = rtree.RTreeCreateTree(-1, 0, 3)
  1216. for i in xrange(10):
  1217. rect = rtree.RTreeAllocRect(tree)
  1218. rtree.RTreeSetRect3D(rect, tree,
  1219. float(i - 2), float(i + 2),
  1220. float(i - 2), float(i + 2),
  1221. float(i - 2), float(i + 2))
  1222. rtree.RTreeInsertRect(rect, i + 1, tree)
  1223. print i + 1
  1224. rtree.RTreePrintRect(rect, 1, tree)
  1225. rect = rtree.RTreeAllocRect(tree)
  1226. rtree.RTreeSetRect3D(rect, tree, 2.0, 7.0, 2.0, 7.0, 2.0, 7.0)
  1227. print "Select"
  1228. rtree.RTreePrintRect(rect, 1, tree)
  1229. list_ = gis.ilist()
  1230. num = vector.RTreeSearch2(tree, rect, byref(list_))
  1231. rtree.RTreeFreeRect(rect)
  1232. # print rectangle ids
  1233. print "Number of overlapping rectangles", num
  1234. for i in xrange(list_.n_values):
  1235. print "id", list_.value[i]
  1236. rtree.RTreeDestroyTree(tree)
  1237. ###############################################################################
  1238. def test_4d_rtree():
  1239. """Testing the rtree ctypes wrapper"""
  1240. tree = rtree.RTreeCreateTree(-1, 0, 4)
  1241. for i in xrange(10):
  1242. # Allocate the boundary
  1243. rect = rtree.RTreeAllocRect(tree)
  1244. rtree.RTreeSetRect4D(rect, tree,
  1245. float(i - 2), float(i + 2),
  1246. float(i - 2), float(i + 2),
  1247. float(i - 2), float(i + 2),
  1248. float(i - 2), float(i + 2))
  1249. rtree.RTreeInsertRect(rect, i + 1, tree)
  1250. rect = rtree.RTreeAllocRect(tree)
  1251. rtree.RTreeSetRect4D(rect, tree, 2.0, 7.0, 2.0,
  1252. 7.0, 2.0, 7.0, 2.0, 7.0)
  1253. list_ = gis.ilist()
  1254. num = vector.RTreeSearch2(tree, rect, byref(list_))
  1255. rtree.RTreeFreeRect(rect)
  1256. # print rectangle ids
  1257. print "Number of overlapping rectangles", num
  1258. for i in xrange(list_.n_values):
  1259. print "id", list_.value[i]
  1260. rtree.RTreeDestroyTree(tree)
  1261. ###############################################################################
  1262. if __name__ == "__main__":
  1263. init()
  1264. test_increment_datetime_by_string()
  1265. test_adjust_datetime_to_granularity()
  1266. test_spatial_extent_intersection()
  1267. test_compute_absolute_time_granularity()
  1268. test_compute_datetime_delta()
  1269. test_spatial_extent_intersection()
  1270. test_spatial_relations()
  1271. test_temporal_topology_builder()
  1272. test_map_list_sorting()
  1273. test_1d_rtree()
  1274. test_2d_rtree()
  1275. test_3d_rtree()
  1276. test_4d_rtree()