def test_update_attr_avg(self):
""" should update arguments correctly with avg operator. """
logging_callback = LoggingCallback()
logging_callback.reward = 0
logging_callback.reward_seen = 2
logging_callback._update_attr('reward', 1, 'avg')
check.equal(logging_callback.reward, 1 / 3)
check.equal(logging_callback.reward_seen, 3)
# With N/A
logging_callback = LoggingCallback()
logging_callback.reward = 'N/A'
logging_callback.reward_seen = 0
logging_callback._update_attr('reward', 1, 'avg')
check.equal(logging_callback.reward, 1)
check.equal(logging_callback.reward_seen, 1)
def test_update_attr_raise(self):
""" should raise ValueError if operator is unknowed. """
with pytest.raises(ValueError, match=r"Unknowed operator.*"):
logging_callback = LoggingCallback()
logging_callback.reward = 'N/A'
logging_callback._update_attr('reward', 1, 'x')
def test_update_attr_last(self):
""" should update arguments correctly with last operator. """
logging_callback = LoggingCallback()
logging_callback.reward = 2
logging_callback._update_attr('reward', 1, 'last')
check.equal(logging_callback.reward, 1)
