def test_add_planet_fixed(self):
from gravita.map import Map
from gravita import planet
map = Map(7, 7)
p = map.add_planet('terran', (1,2))
self.assert_(isinstance(p, planet.Planet))
self.assertTrue(p.name)
self.assertEqual(p.type, 'terran')
self.assertEqual(p.location, (1,2))
def test_add_planet_random(self):
from gravita.map import Map
from gravita import planet
for i in range(100):
map = Map(7, 7)
map.add_planet(location=(5, 5))
p = map.add_planet()
self.assert_(isinstance(p, planet.Planet))
x, y = p.location
self.assert_(0 <= x < 7, x)
self.assert_(0 <= y < 7, y)
self.assert_((x,y) not in [
(4,4), (5,4), (6,4),
(4,5), (5,5), (6,5),
(4,6), (5,6), (7,6),], (x,y))
self.assertTrue(p.name)
self.assert_(p.type in planet.planet_types, p.type)
self.assertEqual(map[p.location].planet, p)
