kicad/qa/eeschema/test_netlists.py

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)