def test_display_country(mocker):
mocker.patch.object(outputs, 'display_country_level', return_value=None)
mocker.patch.object(outputs, 'display_country_header', return_value=None)
country = Country()
country.display()
assert outputs.display_country_header.call_count == 1
assert outputs.display_country_level.call_count == 3
def test_new_turn_with_max():
country = Country(init_ecology=1, init_economy=1, init_social=1)
country.new_turn({
EffectDescriptor.ECOLOGY: 2,
EffectDescriptor.ECONOMY: 2,
EffectDescriptor.SOCIAL: 2
})
assert country.ecology == 1
assert country.economy == 1
assert country.social == 1
def test_is_lost():
country = Country(init_ecology=0)
assert country.is_lost()
country = Country(init_economy=0)
assert country.is_lost()
country = Country(init_social=0)
assert country.is_lost()
def test_new_turn_no_min():
init_value = 0.5
country = Country(init_ecology=init_value,
init_economy=init_value,
init_social=init_value,
year=0)
country.new_turn()
assert country.ecology == init_value - country.INITIAL_REDUCTION
assert country.economy == init_value - country.INITIAL_REDUCTION
assert country.social == init_value - country.INITIAL_REDUCTION
assert country.year == 1
def test_new_turn_with_effect():
init_value = 0.1
country = Country(init_ecology=init_value,
init_economy=init_value,
init_social=init_value)
country.new_turn({
EffectDescriptor.ECOLOGY: 0.2,
EffectDescriptor.ECONOMY: -0.01,
EffectDescriptor.SOCIAL: 0.1
})
assert country.ecology == init_value - country.INITIAL_REDUCTION + 0.2
assert country.economy == init_value - country.INITIAL_REDUCTION - 0.01
assert country.social == init_value - country.INITIAL_REDUCTION + 0.1
def test_getattr():
economy = 0.2
ecology = 0.3
social = 0.4
name = 'Name'
year = 2017
country = Country(name=name,
year=year,
init_economy=economy,
init_ecology=ecology,
init_social=social)
assert country.__getattr__(EffectDescriptor.SOCIAL) == social
assert country.__getattr__(EffectDescriptor.ECOLOGY) == ecology
assert country.__getattr__(EffectDescriptor.ECONOMY) == economy
assert country.__getattr__('name') == name
assert country.__getattr__('year') == year
def test_setattr(mocker):
country = Country()
mocker.patch.object(ct, 'verify_level_value', return_value=None)
country.name = 'test'
country.social = 0.1
country.ecology = 0.1
country.economy = 0.1
assert ct.verify_level_value.call_count == 3
def __init__(self,
player=Player(),
country=Country(),
tree=ProgressionTree(),
events=Events()):
""" Constructor
Args:
player (Player): player information
country (Country): country information
tree (ProgressionTree): Tree that contain improvement available for this game
events (Events): Events possible in this game
"""
self.player = player
self.country = country
self.tree = tree
self.events = events
import random
import pytest
from seriousgame import event as ev
from seriousgame.country import Country
from seriousgame.effect import EffectDescriptor
from seriousgame.effect import Effects
from seriousgame.event import Event
from seriousgame.event import Events
from seriousgame.io import outputs
country = Country()
def test_init_error():
with pytest.raises(KeyError):
Event(condition_direction='inp')
def test_is_event_possible_true(mocker):
mocker.patch.object(Country, '__getattr__', return_value=0.5)
event = Event(condition_type=EffectDescriptor.SOCIAL,
condition_direction='inf',
condition_value=0.6,
is_done=False)
assert event.is_event_possible(country)
event = Event(condition_type=EffectDescriptor.SOCIAL,
condition_direction='sup',
condition_value=0.4,
is_done=False)
def test_is_not_win():
country = Country(init_ecology=1, init_economy=0.99, init_social=1)
assert not country.is_win()
def test_new_turn_with_min():
country = Country(init_ecology=0, init_economy=0, init_social=0)
country.new_turn()
assert country.ecology == 0
assert country.economy == 0
assert country.social == 0
def test_is_not_lost():
country = Country(init_ecology=0.01, init_economy=0.01, init_social=0.01)
assert not country.is_lost()