def test_custom_save_nosave(): """ Test matrices are saved in the savedir with the specified name """ tempdir = tempfile.mkdtemp("_oxasl") try: wsp = Workspace(savedir=tempdir) mat = np.random.rand(4, 4) wsp.set_item("testmat", mat, save_fn=_custom_save, save=False) path = os.path.join(tempdir, "testmat") assert(not os.path.exists(path)) finally: shutil.rmtree(tempdir)
def test_matrix_nosave(): """ Test setting an matrix without saving """ tempdir = tempfile.mkdtemp("_oxasl") try: wsp = Workspace(savedir=tempdir) mat = np.random.rand(4, 4) wsp.set_item("testmat", mat, save=False) path = os.path.join(tempdir, "testmat.mat") assert(not os.path.exists(path)) assert(np.all(mat == wsp.testmat)) finally: shutil.rmtree(tempdir)
def test_image_nosave(): """ Test setting an image without saving """ tempdir = tempfile.mkdtemp("_oxasl") try: wsp = Workspace(savedir=tempdir) img = Image(np.random.rand(5, 5, 5)) wsp.set_item("testimg", img, save=False) path = os.path.join(tempdir, "testimg.nii.gz") assert(not os.path.exists(path)) assert(np.all(img.data == wsp.testimg.data)) finally: shutil.rmtree(tempdir)
def test_custom_save_name(): """ Test matrices are saved in the savedir with the specified name """ tempdir = tempfile.mkdtemp("_oxasl") try: wsp = Workspace(savedir=tempdir) mat = np.random.rand(4, 4) wsp.set_item("testmat", mat, save_name="potato", save_fn=_custom_save) path = os.path.join(tempdir, "testmat") assert(not os.path.exists(path)) path = os.path.join(tempdir, "potato") assert(os.path.exists(path)) with open(path) as sfile: assert("Custom Save" == sfile.read()) finally: shutil.rmtree(tempdir)
def test_image_save_name(): """ Test images are saved in the savedir with the specified name """ tempdir = tempfile.mkdtemp("_oxasl") try: wsp = Workspace(savedir=tempdir) img = Image(np.random.rand(5, 5, 5)) wsp.set_item("testimg", img, save_name="pumpkin") path = os.path.join(tempdir, "testimg.nii.gz") assert(not os.path.exists(path)) path = os.path.join(tempdir, "pumpkin.nii.gz") assert(os.path.isfile(path)) otherimg = Image(path) assert(np.all(img.data == wsp.testimg.data)) assert(np.all(img.data == otherimg.data)) finally: shutil.rmtree(tempdir)
def test_matrix_save_name(): """ Test matrices are saved in the savedir with the specified name """ tempdir = tempfile.mkdtemp("_oxasl") try: wsp = Workspace(savedir=tempdir) mat = np.random.rand(4, 4) wsp.set_item("testmat", mat, save_name="parsnip") path = os.path.join(tempdir, "testmat.mat") assert(not os.path.exists(path)) path = os.path.join(tempdir, "parsnip.mat") assert(os.path.isfile(path)) with open(path) as matfile: othermat = text_to_matrix(matfile.read()) assert(np.all(mat == wsp.testmat)) assert(np.all(mat == othermat)) finally: shutil.rmtree(tempdir)