Geografic-Information-System/lib/python/pygrass/vector/testsuite/test_vector.py

# -*- coding: utf-8 -*-
"""
Created on Wed Jun 18 17:21:42 2014

@author: pietro
"""
from grass.gunittest.case import TestCase
from grass.gunittest.main import test

from grass.pygrass.vector import VectorTopo


class VectorTopoTestCase(TestCase):

    vname = "points_of_interest"

    def test_getitem_slice(self):
        """Test that getitem handle correctly the slice starting from 1"""
        vcoords = ((646341.7386813264, 218873.73056803632),
                   (637772.0990144431, 218842.80557760992))
        with VectorTopo(self.vname, mode="r") as vect:
            coords = tuple([pnt.coords() for pnt in vect[:3]])
            self.assertTupleEqual(vcoords, coords)
            coords = tuple([pnt.coords() for pnt in vect[1:3]])
            self.assertTupleEqual(vcoords, coords)

    def test_getitem_raise(self):
        """Test that getitem raise a value error if the key is not
        an integer or a slice"""
        with VectorTopo(self.vname, mode="r") as vect:
            with self.assertRaises(ValueError):
                vect['value']


if __name__ == '__main__':
    test()