da7009655b63c5d96be0039af421ab56de379ef1,test/test_features.py,TestFeatures,test_pretrain,#TestFeatures#,83

Before Change

            agent, environment = self.prepare(recorder=dict(directory=directory))

            for _ in range(3):
                states = environment.reset()
                terminal = False
                while not terminal:
                    actions = agent.act(states=states)
                    states, terminal, reward = environment.execute(actions=actions)

After Change

            runner.close()

            files = os.listdir(path=directory)
            self.assertEqual(len(files), 10)
            self.assertTrue(
                all(file.startswith("trace-") and file.endswith(".npz") for file in files)
            )

Instances