fixing tests

Signed-off-by: Alessandro Comodi <acomodi@antmicro.com>

tests/test_util.py

@@ -10,6 +10,7 @@ from unittest import TestCase, main
 # in the current subdirectory, which will be a temporary one, to allow concurent
 # testing.
 environ['XRAY_DATABASE_ROOT'] = '.'
+environ['XRAY_PART'] = './'
 
 from prjxray.util import get_roi
 

utils/fasm2frames.py

@@ -109,7 +109,7 @@ def run(
         roi=None,
         debug=False,
         emit_pudc_b_pullup=False):
-    db = Database(db_root)
+    db = Database(db_root, part)
     assembler = fasm_assembler.FasmAssembler(db)
 
     set_features = set()
@@ -124,12 +124,11 @@ def run(
     bank_to_tile = defaultdict(lambda: set())
 
     if part is not None:
-        with open(os.path.join(db_root, part + "_package_pins.csv"),
-                  "r") as fp:
+        with open(os.path.join(db_root, part, "package_pins.csv"), "r") as fp:
             reader = csv.DictReader(fp)
             package_pins = [l for l in reader]
 
-        with open(os.path.join(db_root, part + ".json"), "r") as fp:
+        with open(os.path.join(db_root, part, "part.json"), "r") as fp:
             part_data = json.load(fp)
 
         for bank, loc in part_data["iobanks"].items():

utils/quick_test.py

@@ -1,10 +1,11 @@
 from __future__ import print_function
 import prjxray.db
+import prjxray.util
 import argparse
 
 
-def quick_test(db_root):
-    db = prjxray.db.Database(db_root)
+def quick_test(db_root, part):
+    db = prjxray.db.Database(db_root, part)
     g = db.grid()
 
     # Verify that we have some tile information for every tile in grid.
@@ -47,11 +48,13 @@ def quick_test(db_root):
 def main():
     parser = argparse.ArgumentParser(
         description="Runs a sanity check on a prjxray database.")
-    parser.add_argument('--db-root', required=True)
+
+    util.db_root_arg(parser)
+    util.part_arg(parser)
 
     args = parser.parse_args()
 
-    quick_test(args.db_root)
+    quick_test(args.db_root, args.part)
 
 
 if __name__ == '__main__':

utils/segprint.py

@@ -428,6 +428,7 @@ def load_tiles(db_root):
 
 def run(
         db_root,
+        part,
         bits_file,
         segnames,
         omit_empty_segs=False,
@@ -436,7 +437,7 @@ def run(
         flag_decode_omit=False,
         bit_only=False,
         verbose=False):
-    db = prjxraydb.Database(db_root)
+    db = prjxraydb.Database(db_root, part)
     tiles = load_tiles(db_root)
     segments = mk_segments(tiles)
     bitdata = bitstream.load_bitdata2(open(bits_file, "r"))
@@ -477,6 +478,7 @@ def main():
         description="Decode bits within a tile's address space")
 
     util.db_root_arg(parser)
+    util.part_arg(parser)
     parser.add_argument('--verbose', action='store_true', help='')
     parser.add_argument(
         '-z',
@@ -503,6 +505,7 @@ def main():
 
     run(
         args.db_root,
+        args.part,
         args.bits_file,
         args.segnames,
         args.z,

utils/verify_tile_connections.py

@@ -37,8 +37,8 @@ def make_connection(wires, connection):
         wires[wire] = wire_a_set
 
 
-def make_connections(db_root):
-    db = prjxray.db.Database(db_root)
+def make_connections(db_root, part):
+    db = prjxray.db.Database(db_root, part)
     c = db.connections()
 
     wires = {}
@@ -61,6 +61,7 @@ def read_json5(fname):
 def main():
     parser = argparse.ArgumentParser(
         description="Tests database against raw node list.")
+
     util.db_root_arg(parser)
     util.part_arg(parser)
     parser.add_argument('--raw_node_root', required=True)