graph_tests.py

from graphs import depth_first_connected_nodes_iterative


graph = {
    'A': {
        'B',
    },
    'B': {
        'C',
        'D',
    },
    'C': {
        'E',
    },
    'D': set(),
    'E': set(),
    'F': set(),
}


def test_depth_first_search():
    assert depth_first_connected_nodes_iterative(graph, 'A') == set('ABCDE')