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)