Add support for color in Architecture

test/test_architecture.py

@@ -107,7 +107,7 @@ def test_architecture(schedule_direct_form_iir_lp_filter: Schedule):
                 + pe.entity_name
                 + ' [label="{{<in0> in0|<in1> in1}|'
                 + f'<{pe.entity_name}> {pe.entity_name}'
-                + '|{<out0> out0}}"]\n}'
+                + '|{<out0> out0}}" fillcolor="#00B9E7" style=filled]\n}'
             )
             assert pe._digraph().source in (s, s + '\n')
 
@@ -124,7 +124,7 @@ def test_architecture(schedule_direct_form_iir_lp_filter: Schedule):
         memory.set_entity_name(f"MEM{i}")
         s = (
             'digraph {\n\tnode [shape=record]\n\tMEM0 [label="{{<in0> in0}|<MEM0>'
-            ' MEM0|{<out0> out0}}"]\n}'
+            ' MEM0|{<out0> out0}}" fillcolor="#00CFB5" style=filled]\n}'
         )
         assert memory.schedule_time == 18
         assert memory._digraph().source in (s, s + '\n')