def test_get_friendlies_filter_race(self):
source = create_entity()
friendly_neutral = create_entity()
friendly_demon = create_entity('Voidlord')
combat = SimulatedCombat(([source, friendly_neutral,
friendly_demon], [create_entity()]))
self.assertEqual(
combat.get_friendlies(source, race=hearthstone.enums.Race.DEMON),
[friendly_demon])
def test_get_friendlies(self):
source = create_entity()
friendly = create_entity()
combat = SimulatedCombat(([source, friendly], [create_entity()]))
self.assertEqual(combat.get_friendlies(source), [source, friendly])