Add support for color in Architecture

test/test_architecture.py

@@ -146,6 +146,9 @@ def test_architecture(schedule_direct_form_iir_lp_filter: Schedule):
         processing_elements, memories, direct_interconnects=direct_conn
     )
 
+    assert architecture.direct_interconnects == direct_conn
+
+    # Graph representation
     # Parts are non-deterministic, but this first part seems OK
     s = (
         'digraph {\n\tnode [shape=record]\n\tsplines=spline\n\tsubgraph'
@@ -156,10 +159,11 @@ def test_architecture(schedule_direct_form_iir_lp_filter: Schedule):
     assert architecture._digraph(cluster=False).source.startswith(s)
     assert architecture.schedule_time == 18
 
-    # assert architecture._digraph().source == "foo"
     for pe in processing_elements:
         assert pe.schedule_time == 18
 
+    assert architecture.resource_from_name('adder') == adder
+
 
 def test_move_process(schedule_direct_form_iir_lp_filter: Schedule):
     # Resources