def test_get_current_node_none(self):
visited_nodes = ["a", "b", "c"]
self.assertIsNone(
main.get_current_node(self.labeled_graph, visited_nodes))
def current_node():
# `conftest.py` is imported weirdly by pytest so we cannot just put a
# global variable in it and set it from `Main._ready`
from main import get_current_node
return get_current_node()
def test_get_current_node(self):
visited_nodes = ["b"]
result = "c"
self.assertEqual(
main.get_current_node(self.labeled_graph, visited_nodes), result)