def setUp(self): super(TestEnvironment, self).setUp() self.setup_dir() self.env = Environment(data_dir=self.test_dir, io=MockIOStream(), task=MockTask())