def test_ndata_write_to_then_read_from_temp_file(self): document_model = DocumentModel.DocumentModel() with contextlib.closing(document_model): current_working_directory = os.getcwd() file_path = os.path.join(current_working_directory, "__file.ndata") handler = ImportExportManager.NDataImportExportHandler("ndata1-io-handler", "ndata", ["ndata"]) data_item = DataItem.DataItem(numpy.zeros((16, 16), dtype=numpy.double)) document_model.append_data_item(data_item) display_item = document_model.get_display_item_for_data_item(data_item) handler.write_display_item(None, display_item, file_path, "ndata") self.assertTrue(os.path.exists(file_path)) try: data_items = handler.read_data_items(None, "ndata", file_path) self.assertEqual(len(data_items), 1) data_item = data_items[0] finally: os.remove(file_path)
def test_ndata_write_to_then_read_from_temp_file(self): with TestContext.create_memory_context() as test_context: document_model = test_context.create_document_model() current_working_directory = os.getcwd() file_path = os.path.join(current_working_directory, "__file.ndata") handler = ImportExportManager.NDataImportExportHandler( "ndata1-io-handler", "ndata", ["ndata"]) data_item = DataItem.DataItem( numpy.zeros((16, 16), dtype=numpy.double)) document_model.append_data_item(data_item) display_item = document_model.get_display_item_for_data_item( data_item) handler.write_display_item(display_item, pathlib.Path(file_path), "ndata") self.assertTrue(os.path.exists(file_path)) try: data_items = handler.read_data_items("ndata", pathlib.Path(file_path)) self.assertEqual(len(data_items), 1) data_item = data_items[0] for data_item in data_items: data_item.close() finally: os.remove(file_path)