kicad/qa/testcases/test_002_board_class.py

146 lines
4.3 KiB
Python
Raw Normal View History

import code
import unittest
2014-02-08 23:21:47 +00:00
import os
import pcbnew
import pdb
2014-02-08 23:21:47 +00:00
import tempfile
2014-02-08 00:02:29 +00:00
from pcbnew import *
BACK_COPPER = 'Back_Copper'
B_CU = 'B.Cu'
NEW_NAME = 'My_Fancy_Layer_Name'
class TestBoardClass(unittest.TestCase):
def setUp(self):
2014-02-08 00:02:29 +00:00
self.pcb = LoadBoard("data/complex_hierarchy.kicad_pcb")
2014-02-08 23:21:47 +00:00
self.TITLE="Test Board"
self.COMMENT1="For load/save test"
self.FILENAME=tempfile.mktemp()+".kicad_pcb"
2014-02-08 00:02:29 +00:00
def test_pcb_find_module(self):
module = self.pcb.FindFootprintByReference('P1')
self.assertEqual(module.GetReference(),'P1')
2014-02-08 00:02:29 +00:00
def test_pcb_get_track_count(self):
pcb = BOARD()
2019-05-31 02:30:28 +00:00
self.assertEqual(pcb.Tracks().size(),0)
2014-02-08 00:02:29 +00:00
track0 = PCB_TRACK(pcb)
2014-02-08 00:02:29 +00:00
pcb.Add(track0)
2019-05-31 02:30:28 +00:00
self.assertEqual(pcb.Tracks().size(),1)
2014-02-08 00:02:29 +00:00
track1 = PCB_TRACK(pcb)
2014-02-08 00:02:29 +00:00
pcb.Add(track1)
2019-05-31 02:30:28 +00:00
self.assertEqual(pcb.Tracks().size(),2)
2014-02-08 00:02:29 +00:00
def test_pcb_bounding_box(self):
2014-02-08 00:02:29 +00:00
pcb = BOARD()
track = PCB_TRACK(pcb)
2014-02-08 00:02:29 +00:00
pcb.Add(track)
2022-01-02 20:23:17 +00:00
track.SetStart(VECTOR2I_MM(10.0, 10.0))
track.SetEnd(VECTOR2I_MM(20.0, 30.0))
2014-02-08 00:02:29 +00:00
track.SetWidth(FromMM(0.5))
#!!! THIS FAILS? == 0.0 x 0.0 ??
#height, width = ToMM(pcb.ComputeBoundingBox().GetSize())
bounding_box = pcb.ComputeBoundingBox()
height, width = ToMM(bounding_box.GetSize())
margin = 0 # margin around bounding boxes (currently 0)
self.assertAlmostEqual(width, (30-10) + 0.5 + margin, 2)
self.assertAlmostEqual(height, (20-10) + 0.5 + margin, 2)
2014-02-08 00:02:29 +00:00
def test_pcb_get_pad(self):
pcb = BOARD()
2020-11-13 23:10:45 +00:00
module = FOOTPRINT(pcb)
2014-02-08 00:02:29 +00:00
pcb.Add(module)
2020-11-14 01:06:05 +00:00
pad = PAD(module)
2014-02-08 00:02:29 +00:00
module.Add(pad)
pad.SetShape(PAD_SHAPE_OVAL)
pad.SetSize(VECTOR2I_MM(2.0, 3.0))
2022-01-02 20:23:17 +00:00
pad.SetPosition(VECTOR2I_MM(0,0))
2014-02-08 00:02:29 +00:00
# easy case
2022-01-02 20:23:17 +00:00
p1 = pcb.GetPad(VECTOR2I_MM(0,0))
2014-02-08 00:02:29 +00:00
# top side
2022-01-02 20:23:17 +00:00
p2 = pcb.GetPad(VECTOR2I_MM(0.9,0.0))
2014-02-08 00:02:29 +00:00
# bottom side
2022-01-02 20:23:17 +00:00
p3 = pcb.GetPad(VECTOR2I_MM(0,1.4))
2014-02-08 00:02:29 +00:00
# TODO: get pad == p1 evaluated as true instead
# of relying in the internal C++ object pointer
self.assertEqual(pad.this, p1.this)
self.assertEqual(pad.this, p2.this)
self.assertEqual(pad.this, p3.this)
2014-02-08 23:21:47 +00:00
def test_pcb_save_and_load(self):
pcb = BOARD()
pcb.GetTitleBlock().SetTitle(self.TITLE)
pcb.GetTitleBlock().SetComment(0,self.COMMENT1)
2014-02-08 23:21:47 +00:00
result = SaveBoard(self.FILENAME,pcb)
self.assertTrue(result)
2014-02-08 23:21:47 +00:00
pcb2 = LoadBoard(self.FILENAME)
self.assertNotEqual(pcb2,None)
2014-02-08 23:21:47 +00:00
tb = pcb2.GetTitleBlock()
self.assertEqual(tb.GetTitle(),self.TITLE)
self.assertEqual(tb.GetComment(0),self.COMMENT1)
2014-02-08 23:21:47 +00:00
os.remove(self.FILENAME)
def test_pcb_layer_name_set_get(self):
pcb = BOARD()
pcb.SetLayerName(31, BACK_COPPER)
self.assertEqual(pcb.GetLayerName(31), BACK_COPPER)
def test_pcb_layer_name_set_get(self):
pcb = BOARD()
pcb.SetLayerName(31, BACK_COPPER)
self.assertEqual(pcb.GetLayerName(31), BACK_COPPER)
def test_pcb_layer_id_get(self):
pcb = BOARD()
b_cu_id = pcb.GetLayerID(B_CU)
pcb.SetLayerName(b_cu_id, NEW_NAME)
# ensure we can get the ID for the new name
self.assertEqual(pcb.GetLayerID(NEW_NAME), b_cu_id)
# ensure we can get to the ID via the STD name too
self.assertEqual(pcb.GetLayerID(B_CU), b_cu_id)
def test_footprint_properties(self):
pcb = LoadBoard("data/custom_fields.kicad_pcb")
footprint = pcb.FindFootprintByReference('J1')
expected_properties = {
'Sheet file': 'custom_fields.kicad_sch',
'Sheet name': '',
'myfield': 'myvalue'
}
self.assertEquals(footprint.GetProperties(), expected_properties)
self.assertEquals(footprint.GetProperty('myfield'), 'myvalue')
self.assertEquals(footprint.HasProperty('myfield'), True)
self.assertEquals(footprint.HasProperty('abcd'), False)
footprint.SetProperty('abcd', 'efgh')
self.assertEquals(footprint.HasProperty('abcd'), True)
self.assertEquals(footprint.GetProperty('abcd'), 'efgh')
#def test_interactive(self):
# code.interact(local=locals())
if __name__ == '__main__':
unittest.main()