def test_get_node_then_remove_index_tree(self): fake_tree = FakeTree(FakeNode.make_list(5)) scheduler = SyncScheduler() scheduler.add_index_tree(fake_tree) assert scheduler.get_node() assert scheduler.get_node() scheduler.remove_index_tree(fake_tree) assert scheduler.get_node() is None
def test_remove_index_tree_not_in_scheduler(self): fake_tree = object() scheduler = SyncScheduler() scheduler.remove_index_tree(fake_tree)