def tow_entity(self, request):
self.test_entity1 = Entity(0, 0, 1, 50, 1)
self.test_entity2 = Entity(3, 4, 2, 50, 1)
def teardown():
del self.test_entity1
del self.test_entity2
request.addfinalizer(teardown)
return 'add test_entity1 and test_entity2'
def test_get_in_range(self, one_entity):
test_list = (
Entity(0, 0, 1, 50, 1),
Entity(1, 1, 2, 50, 1),
Entity(2, 2, 3, 50, 2),
Entity(3, 3, 4, 50, 2),
Entity(4, 4, 5, 50, 2)
)
self.test_entity.VisionRange = lambda : 2 #mock
assert self.test_entity.GetInRange(test_list) == {
2: test_list[1],
3: test_list[2]
}
def nextArrival(self):
self.core.increaseEntitiesSystem()
entity = Entity(self.core.getCurrentShift())
if self.outputModule is not None:
self.outputModule.nextArrival(entity)
if self.core.currentTime < Constants.SIMULATION_FINAL_TIME:
arrivalEvent = self.scheduleNextArrival()
self.core.addEvent(arrivalEvent)
def one_entity(self, request):
self.test_entity = Entity(1, 2, 1, 50, 1)
def teardown():
del self.test_entity
request.addfinalizer(teardown)
return 'add test_entity'
def test_init_entity(self):
name = 'entity'
pos = (3, 2)
sprite = 'imgs/dungeon_crawl/monster/angel.png'
entity = Entity(name, pos, sprite)
self.assertEqual(name, entity.name)
self.assertEqual(pos, entity.pos)
self.assertEqual('Entity', str(entity))
self.assertTrue(entity.is_on_pos(pos))
def test_position(self):
name = 'test'
sprite = 'imgs/dungeon_crawl/monster/angel.png'
pos = random_pos()
entity = Entity(name, pos, sprite)
self.assertTrue(entity.is_on_pos(pos))
self.assertTrue(entity.is_on_pos((pos[0] + rd.randint(0, TILE_SIZE), pos[1] + rd.randint(0, TILE_SIZE))))
self.assertFalse(entity.is_on_pos((pos[0] - rd.randint(0, MAIN_WIN_WIDTH), pos[1])))
self.assertFalse(entity.is_on_pos((pos[0], pos[1] - rd.randint(0, MAIN_WIN_HEIGHT))))
def test_name_format(self):
pos = random_pos()
sprite = 'imgs/dungeon_crawl/monster/angel.png'
name = 'test'
entity = Entity(name, pos, sprite)
self.assertEqual('Test', str(entity))
name = 'Test'
entity = Entity(name, pos, sprite)
self.assertEqual('Test', str(entity))
name = 'entity_test'
entity = Entity(name, pos, sprite)
self.assertEqual('Entity Test', str(entity))
name = '5entity_test_01'
entity = Entity(name, pos, sprite)
self.assertEqual('Entity Test', str(entity))
def test_entity_equal(self, one_entity):
self.test_entity1 = Entity(1, 2, 1, 50, 1)
self.test_entity2 = Entity(1, 2, 2, 50, 1)
assert self.test_entity == self.test_entity1
assert self.test_entity != self.test_entity2
