From 497fbcfab5734aac889f15b72384a6a8631fffa4 Mon Sep 17 00:00:00 2001
From: Kaalleen <36401965+kaalleen@users.noreply.github.com>
Date: Mon, 3 Feb 2025 22:37:36 +0100
Subject: Add debug variable to enable sew stack elements (#3476)

---
 tests/test_elements_utils.py | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

(limited to 'tests/test_elements_utils.py')

diff --git a/tests/test_elements_utils.py b/tests/test_elements_utils.py
index bdb78003..4653ac69 100644
--- a/tests/test_elements_utils.py
+++ b/tests/test_elements_utils.py
@@ -1,8 +1,11 @@
-from lib.elements import utils, FillStitch
-from inkex import Rectangle, Group, Style
+from inkex import Group, Rectangle, Style
 from inkex.tester import TestCase
 from inkex.tester.svg import svg
 
+from lib.elements import FillStitch, utils
+
+from .utils import element_count
+
 
 class ElementsUtilsTest(TestCase):
     # These tests test two functions at once, but they're sort of complimentary.
@@ -28,7 +31,7 @@ class ElementsUtilsTest(TestCase):
         }))
 
         elements = utils.nodes_to_elements(utils.iterate_nodes(g))
-        self.assertEqual(len(elements), 2)
+        self.assertEqual(len(elements), element_count())
         self.assertEqual(type(elements[0]), FillStitch)
         self.assertEqual(elements[0].node, rect)
 
@@ -41,7 +44,7 @@ class ElementsUtilsTest(TestCase):
         }))
 
         elements = utils.nodes_to_elements(utils.iterate_nodes(rect))
-        self.assertEqual(len(elements), 2)
+        self.assertEqual(len(elements), element_count())
         self.assertEqual(type(elements[0]), FillStitch)
         self.assertEqual(elements[0].node, rect)
 
-- 
cgit v1.2.3