GlobalNeighbors: tests/test_grid.py comparison

initial commit; half-working prototype

comparison

equal deleted inserted replaced

--1:000000000000
+:5dba84370182
+#!/usr/bin/env python
+"""
+test that we can grid a solution
+"""
+import unittest
+from globalneighbors.grid import LatLonGrid
+class TestGrid(unittest.TestCase):
+"""test gridding functionality"""
+def test_dimensions(self):
+# make a 2 degree grid
+grid = LatLonGrid(90, 180)
+assert grid.n == (90, 180)
+assert grid.d == (2., 2.)
+def test_insertion(self):
+coord = (-23., 122.)
+grid = LatLonGrid(3, 4)
+grid.add(1234, *coord)
+i, j = grid.index(*coord)
+assert i == 1
+assert j == 3
+assert grid[(i,j)] == [1234]
+if __name__ == '__main__':
+unittest.main()

Mercurial > hg > GlobalNeighbors