def test_mine(self):
p = TestMain()
drone = Drone()
drone.pos = vec2d(100, 100)
target = vec2d(100, 100)
drone.target.append(target)
ore = Ore()
ore.pos = vec2d(100, 100)
ore.mine(p, drone)
self.assertEqual(drone.inventory[1][1], 1)
self.assertEqual(ore.quantity, 799999)
def __init__(self):
self.w, self.h = 800, 600
PygameHelper.__init__(self, size=(self.w, self.h), fill=((255,255,255)))
self.hud = Hud()
self.drones = []
for i in range(4):
tempagent = Drone()
tempagent.pos = vec2d(int(uniform(0, self.w - 20)), int(uniform(0, 400)))
target = vec2d(int(uniform(0, self.w - 20)), int(uniform(0, 400)))
tempagent.target.append(target)
self.drones.append(tempagent)
self.selected = self.drones[0]
self.builder = self.drones[0]
self.project = vec2d(0, 0)
self.buildtype = 0
self.buildpos = vec2d(0, 0)
self.buildings = []
self.ores = []
count = 0
while count < 15:
tempore = Ore()
tempore.pos = vec2d(int(uniform(0, 1500)), int(uniform(0, 1500)))
if not len(self.ores) == 0:
flag = 1
for orecheck in self.ores:
dist = orecheck.pos.get_distance(tempore.pos)
if dist < 120:
flag = 0
if flag:
self.ores.append(tempore)
count += 1
else:
self.ores.append(tempore)
count += 1
firstbuild = MainBase()
firstbuild.pos = vec2d(300, 300)
firstbuild.target = vec2d(firstbuild.pos[0] + 10, firstbuild.pos[1])
self.buildings.append(firstbuild)
