FazBrowse GitHub Viewer | Trending |
URL:
| Home
Tools: [Download Repo ZIP]   [Original HTTPS Page]

pytest fix: Files created with NamedTemporary files cannot be opened … · liviolima80/CaffeOnACL@553a645 · GitHub

Repository navigation

Commit 553a645

Browse files
committed
pytest fix: Files created with NamedTemporary files cannot be opened on Windows
1 parent 4ba654f commit 553a645

1 file changed

Lines changed: 9 additions & 9 deletions

File tree

‎python/caffe/test/test_net.py‎

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -173,12 +173,12 @@ class TestLevels(unittest.TestCase):
173173
"""
174174

175175
def setUp(self):
176-
self.f = tempfile.NamedTemporaryFile(mode='w+')
176+
self.f = tempfile.NamedTemporaryFile(mode='w+', delete=False)
177177
self.f.write(self.TEST_NET)
178-
self.f.flush()
178+
self.f.close()
179179

180180
def tearDown(self):
181-
self.f.close()
181+
os.remove(self.f.name)
182182

183183
def check_net(self, net, blobs):
184184
net_blobs = [b for b in net.blobs.keys() if 'data' not in b]
@@ -238,12 +238,12 @@ class TestStages(unittest.TestCase):
238238
"""
239239

240240
def setUp(self):
241-
self.f = tempfile.NamedTemporaryFile(mode='w+')
241+
self.f = tempfile.NamedTemporaryFile(mode='w+', delete=False)
242242
self.f.write(self.TEST_NET)
243-
self.f.flush()
243+
self.f.close()
244244

245245
def tearDown(self):
246-
self.f.close()
246+
os.remove(self.f.name)
247247

248248
def check_net(self, net, blobs):
249249
net_blobs = [b for b in net.blobs.keys() if 'data' not in b]
@@ -320,12 +320,12 @@ class TestAllInOne(unittest.TestCase):
320320
"""
321321

322322
def setUp(self):
323-
self.f = tempfile.NamedTemporaryFile(mode='w+')
323+
self.f = tempfile.NamedTemporaryFile(mode='w+', delete=False)
324324
self.f.write(self.TEST_NET)
325-
self.f.flush()
325+
self.f.close()
326326

327327
def tearDown(self):
328-
self.f.close()
328+
os.remove(self.f.name)
329329

330330
def check_net(self, net, outputs):
331331
self.assertEqual(list(net.blobs['data'].shape), [1,1,10,10])

0 commit comments

Comments
 (0)

Back | FazBrowse Home | New Git URL