From be60145331009433189df107877fed16398af8f8 Mon Sep 17 00:00:00 2001
From: Andrew Bates <andrew.bates@networktocode.com>
Date: Wed, 1 May 2024 13:46:46 -0400
Subject: [PATCH] test: Fixed unit tests

---
 .../tests/testdata/custom_relationship_by_label.yaml            | 2 +-
 .../tests/testdata/custom_relationship_by_slug.yaml             | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/nautobot_design_builder/tests/testdata/custom_relationship_by_label.yaml b/nautobot_design_builder/tests/testdata/custom_relationship_by_label.yaml
index 34c1d209..95e3e2cd 100644
--- a/nautobot_design_builder/tests/testdata/custom_relationship_by_label.yaml
+++ b/nautobot_design_builder/tests/testdata/custom_relationship_by_label.yaml
@@ -48,7 +48,7 @@ checks:
       model: "nautobot.ipam.models.VLAN"
       query: {vid: "43"}
 
-  - equal:
+  - count_equal:
       - model: "nautobot.extras.models.RelationshipAssociation"
         query: {relationship__name: "Device to VLANS"}
         attribute: "destination"
diff --git a/nautobot_design_builder/tests/testdata/custom_relationship_by_slug.yaml b/nautobot_design_builder/tests/testdata/custom_relationship_by_slug.yaml
index fae287b7..8a5f1c04 100644
--- a/nautobot_design_builder/tests/testdata/custom_relationship_by_slug.yaml
+++ b/nautobot_design_builder/tests/testdata/custom_relationship_by_slug.yaml
@@ -48,7 +48,7 @@ checks:
       model: "nautobot.ipam.models.VLAN"
       query: {vid: "43"}
 
-  - equal:
+  - count_equal:
       - model: "nautobot.extras.models.RelationshipAssociation"
         query: {relationship__name: "Device to VLANS"}
         attribute: "destination"