Skip to content

Commit 90c95dd

Browse files
committed
[tests] Update test for frontend
1 parent 31f1869 commit 90c95dd

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

tests/test_network.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,11 @@ def test_hierarchical_only(self):
99
af.examples.arches.networked.hierarchical_only,
1010
)
1111
self.assertIn("PeArray", spec.arch.nodes)
12-
self.assertEqual(spec.arch.nodes["PeArray"].get_fanout(), 4)
12+
self.assertEqual(spec.arch.nodes["PeArray"].get_fanout(), 1)
13+
self.assertIn("Scratchpad", spec.arch.nodes)
14+
self.assertEqual(spec.arch.nodes["Scratchpad"].get_fanout(), 4)
1315
self.assertIn("MacArray", spec.arch.nodes)
14-
self.assertEqual(spec.arch.nodes["MacArray"].get_fanout(), 4)
16+
self.assertEqual(spec.arch.nodes["MacArray"].get_fanout(), 1)
1517

1618
try:
1719
spec = spec.calculate_component_area_energy_latency_leak()
@@ -23,7 +25,7 @@ def test_flat(self):
2325
af.examples.arches.networked.flat,
2426
)
2527
self.assertIn("NoC", spec.arch.nodes)
26-
self.assertEqual(spec.arch.nodes["NoC"].get_fanout(), 16)
28+
self.assertEqual(spec.arch.nodes["NoC"].get_fanout(), 1)
2729
self.assertEqual(
2830
{n.name for n in spec.arch.get_nodes_of_type(af.spec.Leaf)},
2931
{'MainMemory', 'NoC', 'GlobalBuffer', 'RowBuffer', 'ColumnBuffer', 'DistributedBuffer', 'Scratchpad', 'MAC'}

0 commit comments

Comments
 (0)