86 lines
3.1 KiB
Python
86 lines
3.1 KiB
Python
#!/usr/bin/env python3
|
|
|
|
# This program source code file is part of KiCad, a free EDA CAD application.
|
|
#
|
|
# Copyright (C) 2020 Jon Evans <jon@craftyjon.com>
|
|
# Copyright (C) 2020 KiCad Developers, see AUTHORS.txt for contributors.
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify it
|
|
# under the terms of the GNU General Public License as published by the
|
|
# Free Software Foundation, either version 3 of the License, or (at your
|
|
# option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful, but
|
|
# WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
# General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License along
|
|
# with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
|
|
import argparse
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
parser = argparse.ArgumentParser(description='Tests eeschema netlist generation')
|
|
parser.add_argument('--initialize', action='store_true',
|
|
help='Generates the "good" reference netlists for each test case')
|
|
parser.add_argument('binary_dir', default='.')
|
|
|
|
args = parser.parse_args()
|
|
|
|
eeschema = os.path.abspath(os.path.join(args.binary_dir, 'eeschema/eeschema'))
|
|
|
|
print('eeschema binary: {}'.format(eeschema))
|
|
|
|
# The way we discover testcases is simple:
|
|
#
|
|
# The root directory is hardcoded as ./data/netlists
|
|
# Inside that are testcases in folders.
|
|
#
|
|
# Each testcase must have the root sheet named the same as the folder, for example:
|
|
# ./data/netlists/video/video.sch
|
|
#
|
|
# The good netlist will be stored as ./data/netlists/video/video.net
|
|
# The test netlist will be generated as ./data/netlists/video/video_test.net
|
|
|
|
data_dir = os.path.join(os.getcwd(), 'data/netlists')
|
|
|
|
with os.scandir(data_dir) as it:
|
|
for entry in it:
|
|
if entry.is_dir():
|
|
|
|
project = entry.name
|
|
project_dir = os.path.join(data_dir, project)
|
|
|
|
sch_file = os.path.join(project_dir, project + '.kicad_sch')
|
|
good_net_file = os.path.join(project_dir, project + '.net')
|
|
|
|
net_file = good_net_file if args.initialize else os.path.join(
|
|
project_dir, project + '_test.net')
|
|
|
|
if not os.path.exists(good_net_file) and not args.initialize:
|
|
print("FAILED: {} missing good netlist file for comparison".format(project))
|
|
sys.exit(-1)
|
|
|
|
subprocess.run([eeschema, '--netlist', net_file, sch_file], cwd=project_dir)
|
|
|
|
result = subprocess.run(['./netdiff.py', good_net_file, net_file],
|
|
stdout=subprocess.PIPE)
|
|
|
|
if not args.initialize:
|
|
os.remove(net_file)
|
|
|
|
if result.returncode != 0:
|
|
print("FAILED: {} netlist does not match:".format(project))
|
|
print(result.stdout)
|
|
|
|
sys.exit(result.returncode)
|
|
|
|
sys.exit(0)
|