test_v_what_vect.py 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. """
  2. Created on Sun Jun 09 11:42:54 2018
  3. @author: Sanjeet Bhatti
  4. """
  5. from grass.gunittest.case import TestCase
  6. from grass.gunittest.main import test
  7. from grass.gunittest.gmodules import SimpleModule
  8. from grass.script.core import run_command
  9. from grass.script.utils import decode
  10. class TestVWhatVect(TestCase):
  11. """Test v.what.vect script"""
  12. mapName = 'myhospitals'
  13. @classmethod
  14. def setUpClass(cls):
  15. """setup"""
  16. run_command('g.copy', vector='hospitals,myhospitals')
  17. @classmethod
  18. def tearDownClass(cls):
  19. """Remove created vector"""
  20. cls.runModule('g.remove', type='vector', name=cls.mapName, flags='f')
  21. def test_what_vect(self):
  22. """Uploads vector values"""
  23. run_command('v.db.addcolumn', map=self.mapName,
  24. columns="urb_name varchar(25)")
  25. module = SimpleModule('v.what.vect', map=self.mapName,
  26. query_map='urbanarea', column='urb_name',
  27. query_column='NAME')
  28. self.assertModule(module)
  29. m = SimpleModule('v.db.select', map=self.mapName)
  30. self.assertModule(m)
  31. self.assertRegexpMatches(decode(m.outputs.stdout), 'urb_name')
  32. if __name__ == '__main__':
  33. test()