unit_tests.py 47 KB

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