#!/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()