38b45f190d72044731ed0782212f061d974a35a1,tests/unittests/client/test_experiment_client.py,TestBroken,test_interrupted_trial,#TestBroken#,581

Before Change

    def test_interrupted_trial(self):
        Test that interrupted trials are not set to broken
        with create_experiment(config, base_trial) as (cfg, experiment, client):
            trial = client.suggest()
            assert trial.status == "reserved"

            try:
                raise KeyboardInterrupt

After Change

        Test that interrupted trials are not set to broken
        with create_experiment(config, base_trial) as (cfg, experiment, client):
            with pytest.raises(KeyboardInterrupt):
                with client.suggest() as trial:
                    assert trial.status == "reserved"
                    raise KeyboardInterrupt

            assert client._pacemakers == {}
            assert client.get_trial(trial).status == "interrupted"

Instances