def day_06_challenge_01():
f = open(
os.path.dirname(os.path.realpath(__file__)) +
'/input/challenge_01.txt', 'r')
planet = Planet('COM')
planet.initOrbitsFromCode(f.read())
result = planet.getTotalNumberOfOrbits()
return result
def test_planet_init():
planet = Planet('foo')
assert planet.getName() == 'foo'
assert planet.getPlanetsInOwnOrbit() == []
assert planet.getInOrbitOf() == None
planet2 = Planet('ABC')
assert planet2.getName() == 'ABC'
assert len(planet2.getPlanetsInOwnOrbit()) == 0
assert planet2.getInOrbitOf() == None
def test_putinorbitof():
"""
Planet can put itself in orbit of one other planet
"""
planet = Planet('planet')
the_other_planet = Planet('the other planet')
planet.putInOrbitOf(the_other_planet)
assert planet.getInOrbitOf() == the_other_planet
another_planet = Planet('another planet')
planet.putInOrbitOf(another_planet)
assert planet.getInOrbitOf() == another_planet
def test_putinownorbit_linksinbothdirections():
"""
If Planet B is set as putInOrbitOf() of Planet A,
Planet A will putInOwnOrbit() Planet B.
"""
planetA = Planet('A')
planetB = Planet('B')
assert planetA.getPlanetsInOwnOrbit() == []
assert planetB.getInOrbitOf() == None
planetB.putInOrbitOf(planetA)
assert planetA.getPlanetsInOwnOrbit() == [planetB]
assert planetB.getInOrbitOf() == planetA
def test_putinorbitof_linksinbothdirections():
"""
If Planet B is put in orbit of Planet A, Planet A will be put as
isInOrbitOf of Planet B.
"""
planetA = Planet('A')
planetB = Planet('B')
assert planetA.getPlanetsInOwnOrbit() == []
assert planetB.getInOrbitOf() == None
planetA.putInOwnOrbit(planetB)
assert planetA.getPlanetsInOwnOrbit() == [planetB]
assert planetB.getInOrbitOf() == planetA
def test_convertcodetodict():
code = """ \
ABC)DEF
DEF)GHI
ABC)123
"""
expected_result = {
'ABC': ['DEF', '123'],
'DEF': ['GHI']
}
planet = Planet('ABC')
result = planet.convertCodeToDict(code)
assert result == expected_result
def test_initorbitsfromdict():
orbitsDict = {
'ABC': ['DEF', '123'],
'DEF': ['GHI']
}
planet = Planet('ABC')
planet.initOrbitsFromDict(orbitsDict)
assert len(planet.getPlanetsInOwnOrbit()) == 2
assert planet.getPlanetsInOwnOrbit()[0].getName() == 'DEF'
assert planet.getPlanetsInOwnOrbit()[1].getName() == '123'
assert len(planet.getPlanetsInOwnOrbit()[0].getPlanetsInOwnOrbit()) == 1
assert planet.getPlanetsInOwnOrbit()[0].getPlanetsInOwnOrbit()[
0].getName() == 'GHI'
assert len(planet.getPlanetsInOwnOrbit()[1].getPlanetsInOwnOrbit()) == 0
def test_initorbitsfromcode():
code = """ \
ABC)DEF
DEF)GHI
ABC)123
"""
planet = Planet('ABC')
planet.initOrbitsFromCode(code)
assert len(planet.getPlanetsInOwnOrbit()) == 2
assert planet.getPlanetsInOwnOrbit()[0].getName() == 'DEF'
assert planet.getPlanetsInOwnOrbit()[1].getName() == '123'
assert len(planet.getPlanetsInOwnOrbit()[0].getPlanetsInOwnOrbit()) == 1
assert planet.getPlanetsInOwnOrbit()[0].getPlanetsInOwnOrbit()[
0].getName() == 'GHI'
assert len(planet.getPlanetsInOwnOrbit()[1].getPlanetsInOwnOrbit()) == 0
def test_putinownorbit():
"""
Planet can put multiple planets in its own orbit
"""
planet = Planet('planet')
planet2 = Planet('planet 2')
planet3 = Planet('planet 3')
planet4 = Planet('planet 4')
assert planet.getPlanetsInOwnOrbit() == []
planet.putInOwnOrbit(planet2)
assert planet.getPlanetsInOwnOrbit() == [planet2]
planet.putInOwnOrbit(planet3)
assert planet.getPlanetsInOwnOrbit() == [planet2, planet3]
planet.putInOwnOrbit(planet4)
assert planet.getPlanetsInOwnOrbit() == [planet2, planet3, planet4]
def test_gettotalnumberoforbits():
"""
D
/
COM - B - C
"""
code1 = """ \
COM)B
B)C
B)D
"""
COM1 = Planet('COM')
COM1.initOrbitsFromCode(code1)
assert COM1.getTotalNumberOfOrbits() == 5
"""
COM - B - C - D
"""
code2 = """ \
COM)B
B)C
C)D
"""
COM2 = Planet('COM')
COM2.initOrbitsFromCode(code2)
assert COM2.getTotalNumberOfOrbits() == 6
"""
COM
"""
COM3 = Planet('COM')
assert COM3.getTotalNumberOfOrbits() == 0
# # """
# # G - H J - K - L
# # / /
# # COM - B - C - D - E - F
# # \
# # I
# # """
code4 = """ \
COM)B
B)C
B)G
G)H
C)D
D)E
D)I
E)F
E)J
J)K
K)L
"""
COM4 = Planet('COM')
COM4.initOrbitsFromCode(code4)
assert COM4.getTotalNumberOfOrbits() == 42