def test_saveAndLoad(self): # test basic saving a loading functionality # new registration methods should add tests # for loading and saving random.seed(42) ref = random.randn(25, 25) im = shift(ref, [2, 0], mode='constant', order=0) im2 = shift(ref, [0, 2], mode='constant', order=0) imIn = ImagesLoader(self.sc).fromArrays([im, im2]) reg = Registration('crosscorr') reg.prepare(ref) model1 = reg.fit(imIn) t = tempfile.mkdtemp() model1.save(t + '/test.json') # with open(t + '/test.json', 'r') as fp: # print fp.read() model2 = Registration.load(t + '/test.json') # print model2 out1 = model1.transform(imIn).first()[1] out2 = model2.transform(imIn).first()[1] assert_true(allclose(out1, out2))
def test_save_load(self): # test basic saving a loading functionality # new registration methods should add tests # for loading and saving random.seed(42) ref = random.randn(25, 25) im = shift(ref, [2, 0], mode='constant', order=0) imin = ImagesLoader(self.sc).fromArrays(im) reg = Registration('crosscorr') reg.prepare(ref) model1 = reg.fit(imin) t = tempfile.mkdtemp() model1.save(t + '/test.json') model2 = Registration.load(t + '/test.json') out1 = model1.transform(imin).first()[1] out2 = model2.transform(imin).first()[1] assert (allclose(out1, out2))