| FazBrowse GitHub Viewer | Trending | | Home |
| Tools: [Download Repo ZIP] [Original HTTPS Page] |
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -173,12 +173,12 @@ class TestLevels(unittest.TestCase): | |||
| 173 | 173 | """ | |
| 174 | 174 | ||
| 175 | 175 | def setUp(self): | |
| 176 | - self.f = tempfile.NamedTemporaryFile(mode='w+') | ||
| 176 | + self.f = tempfile.NamedTemporaryFile(mode='w+', delete=False) | ||
| 177 | 177 | self.f.write(self.TEST_NET) | |
| 178 | - self.f.flush() | ||
| 178 | + self.f.close() | ||
| 179 | 179 | ||
| 180 | 180 | def tearDown(self): | |
| 181 | - self.f.close() | ||
| 181 | + os.remove(self.f.name) | ||
| 182 | 182 | ||
| 183 | 183 | def check_net(self, net, blobs): | |
| 184 | 184 | net_blobs = [b for b in net.blobs.keys() if 'data' not in b] | |
@@ -238,12 +238,12 @@ class TestStages(unittest.TestCase): | |||
| 238 | 238 | """ | |
| 239 | 239 | ||
| 240 | 240 | def setUp(self): | |
| 241 | - self.f = tempfile.NamedTemporaryFile(mode='w+') | ||
| 241 | + self.f = tempfile.NamedTemporaryFile(mode='w+', delete=False) | ||
| 242 | 242 | self.f.write(self.TEST_NET) | |
| 243 | - self.f.flush() | ||
| 243 | + self.f.close() | ||
| 244 | 244 | ||
| 245 | 245 | def tearDown(self): | |
| 246 | - self.f.close() | ||
| 246 | + os.remove(self.f.name) | ||
| 247 | 247 | ||
| 248 | 248 | def check_net(self, net, blobs): | |
| 249 | 249 | net_blobs = [b for b in net.blobs.keys() if 'data' not in b] | |
@@ -320,12 +320,12 @@ class TestAllInOne(unittest.TestCase): | |||
| 320 | 320 | """ | |
| 321 | 321 | ||
| 322 | 322 | def setUp(self): | |
| 323 | - self.f = tempfile.NamedTemporaryFile(mode='w+') | ||
| 323 | + self.f = tempfile.NamedTemporaryFile(mode='w+', delete=False) | ||
| 324 | 324 | self.f.write(self.TEST_NET) | |
| 325 | - self.f.flush() | ||
| 325 | + self.f.close() | ||
| 326 | 326 | ||
| 327 | 327 | def tearDown(self): | |
| 328 | - self.f.close() | ||
| 328 | + os.remove(self.f.name) | ||
| 329 | 329 | ||
| 330 | 330 | def check_net(self, net, outputs): | |
| 331 | 331 | self.assertEqual(list(net.blobs['data'].shape), [1,1,10,10]) | |
| Back | FazBrowse Home | New Git URL |
0 commit comments