Add __str__ for StarSystem and related tests

Author: mimi1vx

openmoo2/objects/system.py

@@ -70,3 +70,12 @@ def supernova(self):
         self.color = 'black'
         self.__administrator = None
         self.__planets = {x + 1: None for x in range(StarSystem.max_planets)}
+
+    def __str__(self):
+        if self.color == 'black':
+            return "Black hole star system named: {!s}".format(self.name)
+        tmpstring = "{!s} star named {!s} with planets:\n".format(self.color, self.name)
+        tmpstring += "".join(["Orbit {!s}:\n {!s}\n".format(x, y) for x, y in self.planets.items() if y])
+        if self.administrator:
+            tmpstring += "Star system has assigned administrator: {!s}".format(self.administrator)
+        return tmpstring

tests/test_system.py

@@ -93,3 +93,23 @@ def test_supernova(self):
         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')