Fix event chain bg update. (#9118)

We passed in a graph to `sorted_topologically` which didn't have an
entry for each node (as we dropped nodes with no edges).
This commit is contained in:
Erik Johnston 2021-01-14 18:57:32 +00:00 committed by GitHub
parent d2479c6870
commit 1a08e0cdab
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 10 additions and 1 deletions

1
changelog.d/9118.misc Normal file
View File

@ -0,0 +1 @@
Improve efficiency of large state resolutions.

View File

@ -92,7 +92,7 @@ def sorted_topologically(
node = heapq.heappop(zero_degree)
yield node
for edge in reverse_graph[node]:
for edge in reverse_graph.get(node, []):
if edge in degree_map:
degree_map[edge] -= 1
if degree_map[edge] == 0:

View File

@ -56,6 +56,14 @@ class SortTopologically(TestCase):
graph = {} # type: Dict[int, List[int]]
self.assertEqual(list(sorted_topologically([], graph)), [])
def test_handle_empty_graph(self):
"Test that a graph where a node doesn't have an entry is treated as empty"
graph = {} # type: Dict[int, List[int]]
# For disconnected nodes the output is simply sorted.
self.assertEqual(list(sorted_topologically([1, 2], graph)), [1, 2])
def test_disconnected(self):
"Test that a graph with no edges work"