Python OrbitTree.count_direct_orbits Examples

Example #1

File: test_MapNavigator.py Project: roechi/advent_of_code_2019

def test_add_and_count_direct():
    tree = OrbitTree('O')
    tree.add('O', 'A')
    tree.add('O', 'B')
    tree.add('A', 'AA')

    direct_orbits_count = tree.count_direct_orbits()
    assert_equal(direct_orbits_count, 3)

    tree.add('A', 'AAA')
    tree.add('B', 'C')

    direct_orbits_count = tree.count_direct_orbits()
    assert_equal(direct_orbits_count, 5)

Example #2

File: test_MapNavigator.py Project: roechi/advent_of_code_2019

def test_result_part_one():
    t = open('../resources/input_6.txt')
    lines = t.readlines()
    t.close()

    orbit_tupels = list(map(split_to_orbit_tuple, lines))

    tree = OrbitTree('COM')
    build_tree(tree, 'COM', orbit_tupels)

    orbits = tree.count_direct_orbits() + tree.count_indirect_orbits()
    print('The total number of orbits is: ' + str(orbits))

Example #3

File: test_MapNavigator.py Project: roechi/advent_of_code_2019

def test_for_given_map():
    tree = OrbitTree('COM')
    tree.add('COM', 'B')
    tree.add('B', 'G')
    tree.add('G', 'H')
    tree.add('B', 'C')
    tree.add('C', 'D')
    tree.add('D', 'I')
    tree.add('D', 'E')
    tree.add('E', 'J')
    tree.add('J', 'K')
    tree.add('K', 'L')
    tree.add('E', 'F')

    total_orbits = tree.count_direct_orbits() + tree.count_indirect_orbits()
    assert_equal(total_orbits, 42)