kicad/qa/tests/pcbnewswig/test_005_connectivity_data.py

39 lines
1.6 KiB
Python

"""Unit-test parts of the CONNECTIVITY_DATA Python API."""
import pytest
import pcbnew
class TestConnectivity:
"""Test that calls on BOARD.Connectivity() are functional."""
pcb : pcbnew.BOARD = None
def setup_method(self):
"""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."""
connectivity = self.pcb.GetConnectivity()
assert type(connectivity).__name__ == "CONNECTIVITY_DATA"
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()))
track_with_pad = tracks[1]
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_track = list(net_tracks)[0]
connected_tracks = self.connectivity.GetConnectedTracks(net_track)
assert len(connected_tracks) > 1
assert all(track.GetClass() == "PCB_TRACK" for track in connected_tracks)