def test_wrong_input_action_length(game_name, reward_type, num_agent): game = PBeautyGame(num_agent, game_name, reward_type=reward_type) with pytest.raises(WrongActionInputLength) as excinfo: game.step([10] * (num_agent + 1)) assert f"Expected number of actions is {num_agent}" in str(excinfo.value)
def test_create_wrong_name_reward(): with pytest.raises(RewardTypeNotFound) as excinfo: game = PBeautyGame(2, game_name="pbeauty", reward_type="abc") assert "abc" in str(excinfo.value)
def test_create_wrong_game_name(): with pytest.raises(EnvironmentNotFound) as excinfo: game = PBeautyGame(2, game_name="abc") assert "abc" in str(excinfo.value)
def test_create_game(): game = PBeautyGame(2)