cd50ec1c075fed30b19771879b818ba14ecfb816,test/test_features.py,TestFeatures,test_pretrain,#TestFeatures#,85

Before Change

        self.start_tests(name="pretrain")

        // Remove directory if exists
        if os.path.exists(path=self.__class__.directory):
            for filename in os.listdir(path=self.__class__.directory):
                os.remove(path=os.path.join(self.__class__.directory, filename))
            os.rmdir(path=self.__class__.directory)

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

        for _ in range(3):
            states = environment.reset()

After Change

            agent.close()
            environment.close()

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

Instances