def test_star_administrator(self):
star = StarSystem("Terra", "orange")
eq_(star.administrator, None)
star.administrator = "emperor"
eq_(star.administrator, "emperor")
del star.administrator
eq_(star.administrator, None)
def test_string_planets(self):
star = StarSystem("gamma", "gelb")
star.planets = self.planetA
star.planets = self.planetD
star.planets = self.planetB
star.planets = self.planetC
eq_(
str(star),
'gelb star named gamma with planets:\nOrbit 1:\n Planeta : asteroids\nOrbit 3:\n Planeta : giant\nOrbit 4:\n Planeta : planet\n'
)
def test_string_planets_administrator(self):
star = StarSystem("delta", "gelb")
star.planets = self.planetA
star.planets = self.planetD
star.planets = self.planetB
star.planets = self.planetC
star.administrator = "cpt. Kirk"
eq_(
str(star),
'gelb star named delta with planets:\nOrbit 1:\n Planeta : asteroids\nOrbit 3:\n Planeta : giant\nOrbit 4:\n Planeta : planet\nStar system has assigned administrator: cpt. Kirk'
)
def test_star(self):
star = StarSystem("foo", "orange")
eq_(star.color, "orange")
eq_(star.name, "foo")
eq_(star.administrator, None)
eq_(star.planets, {})
ok_('foo' in star._systems)
class TestSystem(object):
planetA = FakePlanet("asteroids")
planetB = FakePlanet("giant")
planetC = FakePlanet("planet")
planetD = None
blackhole = StarSystem("hole", "black")
def test_black_hole(self):
eq_(self.blackhole.name, 'hole')
eq_(self.blackhole.color, "black")
def test_blackhole_planetes(self):
eq_(self.blackhole.planets, {x+1: None for x in range(StarSystem.max_planets)})
@raises(Exception)
def test_add_planet_black_hole(self):
self.blackhole.planets = self.planetD
@raises(Exception)
def test_add_administrator(self):
self.blackhole.administrator = "Pepa z depa"
def test_star(self):
star = StarSystem("foo", "orange")
eq_(star.color, "orange")
eq_(star.name, "foo")
eq_(star.administrator, None)
eq_(star.planets, {})
ok_('foo' in star._systems)
def test_star_add_planet(self):
star = StarSystem("boo", "orange")
star.planets = "one"
eq_(star.planets, {1: 'one'})
@raises(Exception)
def test_star_add_many_planets(self):
star = StarSystem("baz", "orange")
for i in range(StarSystem.max_planets+1):
star.planets = self.planetC
def test_rename_star(self):
star = StarSystem("fii", "orange")
eq_(star.name, 'fii')
star.name = "faa"
eq_(star.name, "faa")
ok_("fii" not in star._systems)
@raises(Exception)
def test_rename_star_wrongly(self):
x = StarSystem._systems[1]
self.blackhole.name = x
def test_star_administrator(self):
star = StarSystem("Terra", "orange")
eq_(star.administrator, None)
star.administrator = "emperor"
eq_(star.administrator, "emperor")
del star.administrator
eq_(star.administrator, None)
def test_supernova(self):
star = StarSystem("beta", "yellow")
star.administrator = "wolf"
star.planets = "asteroids"
star.planets = "planet"
star.planets = "planet"
star.planets = "giant"
eq_(star.planets, {
1: 'asteroids',
2: 'planet',
3: 'planet',
4: 'giant'
})
star.supernova()
eq_(star.administrator, None)
eq_(star.color, 'black')
eq_(star.planets, {x + 1: None for x in range(StarSystem.max_planets)})
def test_delete_name(self):
star = StarSystem("alfa", "blue")
eq_(star.name, 'alfa')
del star.name
eq_(star.name, 'alfa')
def test_rename_star(self):
star = StarSystem("fii", "orange")
eq_(star.name, 'fii')
star.name = "faa"
eq_(star.name, "faa")
ok_("fii" not in star._systems)
def test_star_add_many_planets(self):
star = StarSystem("baz", "orange")
for i in range(StarSystem.max_planets + 1):
star.planets = self.planetC
def test_star_add_delete_planet(self):
star = StarSystem("boo", "orange")
star.planets = "one"
eq_(star.planets, {1: 'one'})
del star.planets
eq_(star.planets, {1: 'one'})
class TestSystem(object):
planetA = FakePlanet("asteroids")
planetB = FakePlanet("giant")
planetC = FakePlanet("planet")
planetD = None
blackhole = StarSystem("hole", "black")
def test_black_hole(self):
eq_(self.blackhole.name, 'hole')
eq_(self.blackhole.color, "black")
def test_blackhole_planetes(self):
eq_(self.blackhole.planets,
{x + 1: None
for x in range(StarSystem.max_planets)})
@raises(Exception)
def test_add_planet_black_hole(self):
self.blackhole.planets = self.planetD
@raises(Exception)
def test_add_administrator(self):
self.blackhole.administrator = "Pepa z depa"
def test_star(self):
star = StarSystem("foo", "orange")
eq_(star.color, "orange")
eq_(star.name, "foo")
eq_(star.administrator, None)
eq_(star.planets, {})
ok_('foo' in star._systems)
def test_star_add_delete_planet(self):
star = StarSystem("boo", "orange")
star.planets = "one"
eq_(star.planets, {1: 'one'})
del star.planets
eq_(star.planets, {1: 'one'})
@raises(Exception)
def test_star_add_many_planets(self):
star = StarSystem("baz", "orange")
for i in range(StarSystem.max_planets + 1):
star.planets = self.planetC
def test_rename_star(self):
star = StarSystem("fii", "orange")
eq_(star.name, 'fii')
star.name = "faa"
eq_(star.name, "faa")
ok_("fii" not in star._systems)
@raises(Exception)
def test_rename_star_wrongly(self):
x = StarSystem._systems[1]
self.blackhole.name = x
def test_star_administrator(self):
star = StarSystem("Terra", "orange")
eq_(star.administrator, None)
star.administrator = "emperor"
eq_(star.administrator, "emperor")
del star.administrator
eq_(star.administrator, None)
def test_delete_name(self):
star = StarSystem("alfa", "blue")
eq_(star.name, 'alfa')
del star.name
eq_(star.name, 'alfa')
def test_supernova(self):
star = StarSystem("beta", "yellow")
star.administrator = "wolf"
star.planets = "asteroids"
star.planets = "planet"
star.planets = "planet"
star.planets = "giant"
eq_(star.planets, {
1: 'asteroids',
2: 'planet',
3: 'planet',
4: 'giant'
})
star.supernova()
eq_(star.administrator, None)
eq_(star.color, 'black')
eq_(star.planets, {x + 1: None for x in range(StarSystem.max_planets)})
def test_string_blackhole(self):
eq_(str(self.blackhole), "Black hole star system named: hole")
def test_string_planets(self):
star = StarSystem("gamma", "gelb")
star.planets = self.planetA
star.planets = self.planetD
star.planets = self.planetB
star.planets = self.planetC
eq_(
str(star),
'gelb star named gamma with planets:\nOrbit 1:\n Planeta : asteroids\nOrbit 3:\n Planeta : giant\nOrbit 4:\n Planeta : planet\n'
)
def test_string_planets_administrator(self):
star = StarSystem("delta", "gelb")
star.planets = self.planetA
star.planets = self.planetD
star.planets = self.planetB
star.planets = self.planetC
star.administrator = "cpt. Kirk"
eq_(
str(star),
'gelb star named delta with planets:\nOrbit 1:\n Planeta : asteroids\nOrbit 3:\n Planeta : giant\nOrbit 4:\n Planeta : planet\nStar system has assigned administrator: cpt. Kirk'
)