| 205 | shutil.rmtree(cls.class_tmp_dir) |
| 206 | |
| 207 | def setUp(self): |
| 208 | # Make sure this file is not being run from the source directory, to |
| 209 | # avoid picking up source files. |
| 210 | if os.path.isdir( |
| 211 | os.path.join(os.path.dirname(__file__), 'tensorflowjs')): |
| 212 | self.fail('Do not run this test from the Python source directory. ' |
| 213 | 'This file is intended to be run on pip install.') |
| 214 | |
| 215 | self._tmp_dir = tempfile.mkdtemp() |
| 216 | super(APIAndShellTest, self).setUp() |
| 217 | |
| 218 | def tearDown(self): |
| 219 | if os.path.isdir(self._tmp_dir): |