提交 1c07a240 authored 作者: Olivier Delalleau's avatar Olivier Delalleau

Tests do not sneakily delete files obj.save and objects.save, nor do they leave them behind

上级 5e16fbee
""" test code snippet in the Theano tutorials. """ test code snippet in the Theano tutorials.
""" """
import unittest import os, unittest
import theano import theano
import theano.tensor as T import theano.tensor as T
from theano import function from theano import function
...@@ -722,6 +722,15 @@ class T_loading_and_saving(unittest.TestCase): ...@@ -722,6 +722,15 @@ class T_loading_and_saving(unittest.TestCase):
mode_instance = theano.compile.mode.get_mode(None) mode_instance = theano.compile.mode.get_mode(None)
if not isinstance(mode_instance, theano.compile.debugmode.DebugMode): if not isinstance(mode_instance, theano.compile.debugmode.DebugMode):
if os.path.exists('obj.save') or os.path.exists('objects.save'):
# We do not want to delete these files silently, in case for
# some reason they would be something else than test-generated
# files.
# Ideally we would save those files in a temporary directory...
raise AssertionError(
'Please get rid of files obj.save and '
'objects.save in directory %s' % os.getcwd())
f = file('obj.save', 'wb') f = file('obj.save', 'wb')
cPickle.dump(my_obj, f, protocol=cPickle.HIGHEST_PROTOCOL) cPickle.dump(my_obj, f, protocol=cPickle.HIGHEST_PROTOCOL)
f.close() f.close()
...@@ -746,6 +755,9 @@ class T_loading_and_saving(unittest.TestCase): ...@@ -746,6 +755,9 @@ class T_loading_and_saving(unittest.TestCase):
loaded_objects.append(cPickle.load(f)) loaded_objects.append(cPickle.load(f))
f.close() f.close()
# Cleanup created files.
os.remove('obj.save')
os.remove('objects.save')
class T_modes(unittest.TestCase): class T_modes(unittest.TestCase):
## All tests here belog to ## All tests here belog to
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论