mirror of https://github.com/openXC7/prjxray.git
75 lines
2.1 KiB
Python
Executable File
75 lines
2.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# Copyright (C) 2017-2020 The Project X-Ray Authors.
|
|
#
|
|
# Use of this source code is governed by a ISC-style
|
|
# license that can be found in the LICENSE file or at
|
|
# https://opensource.org/licenses/ISC
|
|
#
|
|
# SPDX-License-Identifier: ISC
|
|
|
|
from os import environ, getcwd, chdir
|
|
import json
|
|
from tempfile import TemporaryDirectory
|
|
from contextlib import contextmanager
|
|
from unittest import TestCase, main
|
|
|
|
# Setup location of database file to a relative term so it can be generated
|
|
# 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
|
|
|
|
|
|
@contextmanager
|
|
def setup_database(contents):
|
|
with TemporaryDirectory() as d:
|
|
olddir = getcwd()
|
|
chdir(d)
|
|
e = None
|
|
with open('tilegrid.json', 'w') as fd:
|
|
json.dump(contents, fd)
|
|
try:
|
|
yield
|
|
except Exception as ereal:
|
|
e = ereal
|
|
chdir(olddir)
|
|
if e is not None:
|
|
raise e
|
|
|
|
|
|
class TestUtil(TestCase):
|
|
def test_get_roi_gen_sites(self):
|
|
makedb = lambda sites: {
|
|
"ATILE": {
|
|
"bits": {
|
|
"CLB_IO_CLK": {
|
|
"baseaddr": "0x00400F00",
|
|
"frames": 28,
|
|
"height": 2,
|
|
"offset": 0,
|
|
"words": 2
|
|
}
|
|
},
|
|
"grid_x": 10,
|
|
"grid_y": 10,
|
|
"segment": "ASEGMENT",
|
|
"segment_type": "bram0_l",
|
|
"sites": sites,
|
|
"prohibited_sites": [],
|
|
"type": "BRAM_INT_INTERFACE_L"
|
|
}
|
|
}
|
|
with setup_database(makedb({})):
|
|
self.assertListEqual(list(get_roi().gen_sites()), [])
|
|
with setup_database(makedb({'FOO': 'BAR'})):
|
|
self.assertListEqual(
|
|
list(get_roi().gen_sites()), [('ATILE', 'FOO', 'BAR')])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|