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)