diff --git a/script/test/TestLayout.py b/script/test/TestLayout.py index 7cf914b..78b1ce6 100644 --- a/script/test/TestLayout.py +++ b/script/test/TestLayout.py @@ -31,8 +31,9 @@ def plot_base_plate(points = None, title = None): for point in points: c = Color.GRAY for puck in get_pucks(): - if puck.match(point[0], point[1]) is not None: - c = Color.BLACK + match = puck.match(point[0], point[1]) + if match is not None: + c = Color.CYAN if match == "minispine" else Color.GREEN plot_cross(p,point[0], point[1], size = 10, width = 1, color = c, name = "P"+ str(points.index(point))) #plot_point(p,point[0], point[1], size = 5, color = Color.BLACK, name = "Pc"+ str(points.index(point)))