import pytest import os.path import sys from ee.kicad.bom import * from ee.kicad.bom.io import read_bom basedir = os.path.dirname(os.path.abspath(__file__)) @pytest.mark.parametrize("s, ref, val, rest", [ ("C12", "C", 12, ""), ("C12n", "C", 12, "n"), ("C", "C", sys.maxsize, ""), ("Foo", "Foo", sys.maxsize, ""), ("+3.0VA1", "+3.0VA1", sys.maxsize, ""), ]) def test_split_ref(s, ref, val, rest): assert split_ref(s) == (ref, val, rest) def test_read_bom_1(): bom = read_bom(basedir + '/../demo/kicad/bom/A64-OlinuXino_Rev_C.xml') assert len(bom.get_components()) == 425 def test_read_bom_2(): bom = read_bom(basedir + '/../demo/kicad/bom/gw.xml') assert len(bom.get_components()) == 165 r5 = bom.get_component("R5") assert r5.ref == "R5" assert r5.value == "R0402_100R" assert r5["value"] == "R0402_100R" assert r5.footprint == "Resistors_SMD:R_0402" assert r5.library.name == "gw-cache" assert len(r5.fields) == 4 assert r5.fields["Part Number"] == "CRCW0402100RFKED" assert r5["Part Number"] == "CRCW0402100RFKED" assert set(['ref', 'value', 'Capacitance', 'Color', 'Description', 'Frequency', 'Impedance', 'Inductance', 'Manufacturer', 'Part Number', 'Resistance']) == bom.all_field_names() assert not "foo" in r5 with pytest.raises(KeyError): r5["foo"] def test_read_bom_2_pandas(): bom = read_bom(basedir + '/../demo/kicad/bom/gw.xml').to_pandas() assert len(bom) == 165 print("bom") print(str(bom)) r5 = bom.loc["R5"] print(str(r5.index)) # assert r5.index == "R5" assert r5["ref"] == "R5" assert r5["value"] == "R0402_100R" assert r5["value"] == "R0402_100R" # assert r5["footprint"] == "Resistors_SMD:R_0402" # assert r5.library.name == "gw-cache" # assert len(r5.fields) == 4 # assert r5.fields["Part Number"] == "CRCW0402100RFKED" assert r5["Part Number"] == "CRCW0402100RFKED" # assert set(['ref', 'value', 'Capacitance', 'Color', 'Description', 'Frequency', 'Impedance', 'Inductance', 'Manufacturer', 'Part Number', 'Resistance']) == bom.all_field_names() assert not "foo" in r5 with pytest.raises(KeyError): r5["foo"]