Index: tools/ignition/bytecode_dispatches_report_test.py |
diff --git a/tools/ignition/bytecode_dispatches_report_test.py b/tools/ignition/bytecode_dispatches_report_test.py |
index 523dac2960d8da5e0d2ca9f31288e6201a80b041..9be19e7f6312c49169beeb0c63bd7f3d49dda151 100644 |
--- a/tools/ignition/bytecode_dispatches_report_test.py |
+++ b/tools/ignition/bytecode_dispatches_report_test.py |
@@ -42,13 +42,21 @@ class BytecodeDispatchesReportTest(unittest.TestCase): |
('b', 5) |
]) |
- def test_find_top_dispatch_sources(self): |
- top_dispatch_sources = bdr.find_top_dispatch_sources({ |
- "a": {"a": 10, "b": 8, "c": 7}, |
+ def test_find_top_dispatch_sources_and_destinations(self): |
+ d = { |
+ "a": {"a": 4, "b": 2, "c": 4}, |
"b": {"a": 1, "c": 4}, |
- "c": {"a": 42, "b": 12, "c": 99} |
- }, "b", 10) |
- self.assertListEqual(top_dispatch_sources, [ |
- ("c", 12), |
- ("a", 8) |
+ "c": {"a": 40, "b": 10, "c": 50} |
+ } |
+ top_sources, top_dests = bdr.find_top_dispatch_sources_and_destinations( |
+ d, "b", 10, False) |
+ self.assertListEqual(top_sources, [ |
+ ("c", 10, 0.1), |
+ ("a", 2, 0.2) |
+ ]) |
+ top_sources, top_dests = bdr.find_top_dispatch_sources_and_destinations( |
+ d, "b", 10, True) |
+ self.assertListEqual(top_sources, [ |
+ ("a", 2, 0.2), |
+ ("c", 10, 0.1) |
]) |