b12b05da8ff77022c46cb005657ca985cc029f55,tests/storage_tests/test_rdb.py,TestRDBStorage,test_set_trial_state,#TestRDBStorage#,193

Before Change

        study_id = storage.create_new_study_id()
        trial_id = storage.create_new_trial_id(study_id)

        result_1 = storage.session.query(Trial).filter(Trial.trial_id == trial_id).one().state

        storage.set_trial_state(trial_id, trial_module.State.PRUNED)

        result_2 = storage.session.query(Trial).filter(Trial.trial_id == trial_id).one().state

        assert result_1 == trial_module.State.RUNNING
        assert result_2 == trial_module.State.PRUNED

        storage.close()

    def test_set_trial_param(self):
        // type: () -> None

After Change

    def test_set_trial_state(self):
        // type: () -> None

        storage = self.create_test_storage()
        session = storage.scoped_session()

        study_id = storage.create_new_study_id()
        trial_id = storage.create_new_trial_id(study_id)

Instances