def test_pickling_file_custom_path(self): modelcube = ModelCube(filename=self.filename) test_path = '~/test.mpf' path = modelcube.save(path=test_path) self._files_created.append(path) self.assertTrue(os.path.exists(path)) self.assertEqual(path, os.path.realpath(os.path.expanduser(test_path))) modelcube_restored = ModelCube.restore(path, delete=True) self.assertEqual(modelcube_restored.data_origin, 'file') self.assertIsInstance(modelcube_restored, ModelCube) self.assertIsNotNone(modelcube_restored.data) self.assertFalse(os.path.exists(path))
def test_pickling_api(self, temp_scratch, galaxy): modelcube = ModelCube(plateifu=galaxy.plateifu, bintype=galaxy.bintype, mode='remote') assert modelcube.data_origin == 'api' assert isinstance(modelcube, ModelCube) assert modelcube.data is None file = temp_scratch.join('test_modelcube_api.mpf') modelcube.save(str(file)) assert file.check() is True modelcube = None assert modelcube is None modelcube_restored = ModelCube.restore(str(file)) assert modelcube_restored.data_origin == 'api' assert isinstance(modelcube_restored, ModelCube) assert modelcube_restored.data is None
def test_pickling_file_custom_path(self, temp_scratch, galaxy): modelcube = ModelCube(filename=galaxy.modelpath, bintype=galaxy.bintype) assert modelcube.data_origin == 'file' assert isinstance(modelcube, ModelCube) assert modelcube.data is not None file = temp_scratch.join('mcpickle').join('test_modelcube.mpf') assert file.check(file=1) is False path = modelcube.save(path=str(file)) assert file.check() is True assert os.path.exists(path) modelcube_restored = ModelCube.restore(str(file), delete=True) assert modelcube_restored.data_origin == 'file' assert isinstance(modelcube_restored, ModelCube) assert modelcube_restored.data is not None assert not os.path.exists(path)
def test_pickling_file(self, temp_scratch, galaxy): modelcube = ModelCube(filename=galaxy.modelpath, bintype=galaxy.bintype) assert modelcube.data_origin == 'file' assert isinstance(modelcube, ModelCube) assert modelcube.data is not None file = temp_scratch.join('test_modelcube.mpf') modelcube.save(str(file)) assert file.check() is True assert modelcube.data is not None modelcube = None assert modelcube is None modelcube_restored = ModelCube.restore(str(file)) assert modelcube_restored.data_origin == 'file' assert isinstance(modelcube_restored, ModelCube) assert modelcube_restored.data is not None
def test_pickling_api(self): modelcube = ModelCube(plateifu=self.plateifu, mode='remote') self.assertEqual(modelcube.data_origin, 'api') self.assertIsInstance(modelcube, ModelCube) self.assertIsNone(modelcube.data) path = modelcube.save() self._files_created.append(path) self.assertTrue(os.path.exists(path)) self.assertEqual(os.path.realpath(path), os.path.realpath(self.filename[0:-7] + 'mpf')) modelcube = None self.assertIsNone(modelcube) modelcube_restored = ModelCube.restore(path) self.assertEqual(modelcube_restored.data_origin, 'api') self.assertIsInstance(modelcube_restored, ModelCube) self.assertIsNone(modelcube_restored.data) self.assertEqual(modelcube_restored.header['VERSDRP3'], 'v2_0_1')
def test_pickling_file(self): modelcube = ModelCube(filename=self.filename) self.assertEqual(modelcube.data_origin, 'file') self.assertIsInstance(modelcube, ModelCube) self.assertIsNotNone(modelcube.data) path = modelcube.save() self._files_created.append(path) self.assertTrue(os.path.exists(path)) self.assertEqual(os.path.realpath(path), os.path.realpath(self.filename[0:-7] + 'mpf')) self.assertIsNotNone(modelcube.data) modelcube = None self.assertIsNone(modelcube) modelcube_restored = ModelCube.restore(path) self.assertEqual(modelcube_restored.data_origin, 'file') self.assertIsInstance(modelcube_restored, ModelCube) self.assertIsNotNone(modelcube_restored.data)