diff --git a/tests/test_util.py b/tests/test_util.py index 8ea851ed..dae5b13c 100755 --- a/tests/test_util.py +++ b/tests/test_util.py @@ -21,8 +21,10 @@ from unittest import TestCase, main environ['XRAY_DATABASE_ROOT'] = '.' environ['XRAY_PART'] = './' -from prjxray.util import get_roi - +from prjxray.util import get_roi, get_db_root +from prjxray.db import Database +from prjxray.overlay import Overlay +from prjxray.grid_types import GridLoc @contextmanager def setup_database(contents): @@ -69,6 +71,17 @@ class TestUtil(TestCase): self.assertListEqual( list(get_roi().gen_sites()), [('ATILE', 'FOO', 'BAR')]) + def test_in_roi_overlay(self): + db = Database(get_db_root(), './database/artix7/xc7a50tfgg484-1') + region_dict = {} + region_dict['pr1'] = (10, 58, 0, 51) + region_dict['pr2'] = (10, 58, 52, 103) + overlay = Overlay(db.grid(), region_dict) + self.assertFalse(overlay.tile_in_roi(GridLoc(18, 50))) + self.assertFalse(overlay.tile_in_roi(GridLoc(18, 84))) + self.assertTrue(overlay.tile_in_roi(GridLoc(8, 50))) + self.assertTrue(overlay.tile_in_roi(GridLoc(18, 112))) + self.assertTrue(overlay.tile_in_roi(GridLoc(80, 40))) if __name__ == '__main__': main()