def test_get_current_node_none(self):
     visited_nodes = ["a", "b", "c"]
     self.assertIsNone(
         main.get_current_node(self.labeled_graph, visited_nodes))
Example #2
0
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)