71 lines
2.0 KiB
Python
71 lines
2.0 KiB
Python
"""
|
|
Tests for the KiCost output.
|
|
|
|
For debug information use:
|
|
pytest-3 --log-cli-level debug
|
|
"""
|
|
|
|
import os.path as op
|
|
import sys
|
|
# Look for the 'utils' module from where the script is running
|
|
prev_dir = op.dirname(op.dirname(op.abspath(__file__)))
|
|
if prev_dir not in sys.path:
|
|
sys.path.insert(0, prev_dir)
|
|
# Utils import
|
|
from utils import context
|
|
import logging
|
|
import subprocess
|
|
|
|
|
|
OUT_DIR = 'KiCost'
|
|
|
|
|
|
def convert2csv(xlsx, skip_empty=False, sheet=None):
|
|
csv = xlsx[:-4]+'csv'
|
|
logging.debug('Converting to CSV')
|
|
cmd = ['xlsx2csv']
|
|
if skip_empty:
|
|
cmd.append('--skipemptycolumns')
|
|
if sheet:
|
|
cmd.extend(['-n', sheet])
|
|
cmd.append(xlsx)
|
|
p1 = subprocess.Popen(cmd, stdout=subprocess.PIPE)
|
|
with open(csv, 'w') as f:
|
|
p2 = subprocess.Popen(['egrep', '-i', '-v', r'(\$ date|created|kicost|Total purchase)'], stdin=p1.stdout, stdout=f)
|
|
p2.communicate()[0]
|
|
|
|
|
|
def check_simple(ctx, variant):
|
|
if variant:
|
|
variant = '_'+variant
|
|
name = op.join(OUT_DIR, 'simple'+variant+'.xlsx')
|
|
ctx.expect_out_file(name)
|
|
xlsx = ctx.get_out_path(name)
|
|
convert2csv(xlsx, skip_empty=True)
|
|
ctx.compare_txt(name[:-4]+'csv')
|
|
|
|
|
|
def test_kicost_simple(test_dir):
|
|
prj = 'kibom-variant_kicost'
|
|
ctx = context.TestContextSCH(test_dir, 'test_kicost_simple', prj, 'kicost_simple', OUT_DIR)
|
|
ctx.run()
|
|
check_simple(ctx, '')
|
|
check_simple(ctx, 'default')
|
|
check_simple(ctx, 'production')
|
|
check_simple(ctx, 'test')
|
|
ctx.clean_up()
|
|
|
|
|
|
def test_kicost_bom_simple(test_dir):
|
|
prj = 'kibom-variant_2c'
|
|
ctx = context.TestContextSCH(test_dir, 'test_kicost_bom_simple', prj, 'int_bom_kicost_simple_xlsx', OUT_DIR)
|
|
ctx.run(kicost=True) # , extra_debug=True
|
|
output = op.join(OUT_DIR, prj+'-bom.xlsx')
|
|
ctx.expect_out_file(output)
|
|
convert2csv(ctx.get_out_path(output), sheet='Costs')
|
|
csv = output[:-4]+'csv'
|
|
ctx.compare_txt(csv)
|
|
convert2csv(ctx.get_out_path(output), sheet='Costs (DNF)')
|
|
ctx.compare_txt(csv, output[:-5]+'_dnf.csv')
|
|
ctx.clean_up()
|