Bug fix for stacked systems (#29)

pyxdsm/XDSM.py

@@ -325,7 +325,7 @@ def _build_node_grid(self):
                 style += ',faded'
 
             label = _parse_label(comp.label, comp.label_width)
-            node = node_str.format(style=comp.style, node_name=comp.node_name, node_label=label)
+            node = node_str.format(style=style, node_name=comp.node_name, node_label=label)
             grid[i_row, j_col] = node
 
             row_idx_map[comp.node_name] = i_row

pyxdsm/tests/test_xdsm.py

@@ -3,6 +3,13 @@
 from pyxdsm.XDSM import XDSM, __file__
 from numpy.distutils.exec_command import find_executable
 
+
+def filter_lines(lns):
+    # Empty lines are excluded.
+    # Leading and trailing whitespaces are removed
+    # Comments are removed.
+    return [ln.strip() for ln in lns if ln.strip() and not ln.strip().startswith('%')]
+
 class TestXDSM(unittest.TestCase):
 
     def setUp(self):
@@ -102,6 +109,24 @@ def test_options(self):
         self.assertTrue(os.path.isfile(os.path.join(spec_dir, 'F.json')))
         self.assertTrue(os.path.isfile(os.path.join(spec_dir, 'G_spec.json')))
 
+
+    def test_stacked_system(self): 
+
+        x = XDSM()
+
+        x.add_system('test', 'Optimization', r'\text{test}', stack=True)
+
+        file_name = "stacked_test"
+        x.write(file_name)
+
+        tikz_file = file_name + '.tikz'
+        with open(tikz_file, "r") as f:
+            tikz = f.read()
+
+        self.assertIn(r"\node [Optimization,stack]", tikz)
+
+
+
     def test_tikz_content(self):
         # Check if TiKZ file was created.
         # Compare the content of the sample below and the newly created TiKZ file.
@@ -222,12 +247,6 @@ def test_tikz_content(self):
             
             \end{tikzpicture}"""
 
-        def filter_lines(lns):
-            # Empty lines are excluded.
-            # Leading and trailing whitespaces are removed
-            # Comments are removed.
-            return [ln.strip() for ln in lns if ln.strip() and not ln.strip().startswith('%')]
-
         filename = 'xdsm_test_tikz'
 
         x = XDSM(use_sfmath=True)