Tests fix

Author: tanner-andrulis

tests/vibe_see_readme_in_this_dir/test_arch_flattening.py

@@ -23,6 +23,7 @@
     Toll,
     Leaf,
     Branch,
+    FlattenedArch,
 )
 from accelforge.frontend.workload import Workload
 
@@ -321,7 +322,7 @@ def test_specific_compute_node(self):
         spec = self._make_spec()
         result = spec._get_flattened_architecture(compute_node="MAC")
         # Returns a single path (flat list of Leaf nodes), not list of lists
-        self.assertIsInstance(result, list)
+        self.assertIsInstance(result, FlattenedArch)
         self.assertIsInstance(result[-1], Leaf)
         self.assertEqual(result[-1].name, "MAC")
 
@@ -346,7 +347,7 @@ def test_fork_arch_specific_mac(self):
         spec = self._make_spec(arch=_fork_arch())
         result = spec._get_flattened_architecture(compute_node="MAC")
         # Returns a single path, not list of lists
-        self.assertIsInstance(result, list)
+        self.assertIsInstance(result, FlattenedArch)
         self.assertIsInstance(result[-1], Leaf)
         self.assertEqual(result[-1].name, "MAC")