Update connectivity python test

Previous test depended on which track was first.  Re-saving the board
changed the track order, so there was no pad connected.  It was still in
the connectivity database, just not at the first track
This commit is contained in:
Seth Hillbrand 2023-07-20 15:51:17 -07:00
parent e4b262d6e1
commit 2485e9a38a
1 changed files with 4 additions and 5 deletions

View File

@ -12,7 +12,6 @@ class TestConnectivity:
"""Setup shared attributes."""
self.pcb = pcbnew.LoadBoard("../data/pcbnew/complex_hierarchy.kicad_pcb")
self.connectivity = self.pcb.GetConnectivity()
self.nets = self.pcb.GetNetsByName()
def test_get_connectivity_returns_connectivity_data_object(self):
"""Verify: GetConnectivity() returns a CONNECTIVITY_DATA object."""
@ -21,18 +20,18 @@ class TestConnectivity:
def test_get_connected_pads_on_track_returns_iterable_of_pads(self):
"""Verify: GetConnectedPads(track) returns an iterable of pads."""
net = list(self.nets.values())[1]
tracks = list(self.pcb.TracksInNet(net.GetNetCode()))
tracks = list(self.pcb.TracksInNet(self.pcb.GetNetcodeFromNetname("/12Vext")))
track_with_pad = tracks[1]
"""Note that this returns just the elements directly connected, not everything in the net"""
pads = self.connectivity.GetConnectedPads(track_with_pad)
assert len(pads) > 0
assert all(pad.GetClass() == "PAD" for pad in pads)
def test_get_connected_tracks_returns_iterable_of_tracks(self):
"""Verify: GetConnectedTracks(track) returns an iterable of tracks."""
net = list(self.nets.values())[1]
net_tracks = self.pcb.TracksInNet(net.GetNetCode())
net_tracks = self.pcb.TracksInNet(self.pcb.GetNetcodeFromNetname("/12Vext"))
net_track = list(net_tracks)[0]
"""Note that this returns just the elements directly connected, not everything in the net"""
connected_tracks = self.connectivity.GetConnectedTracks(net_track)
assert len(connected_tracks) > 1
assert all(track.GetClass() == "PCB_TRACK" for track in connected_tracks)