def test_importing_large_numpy_file_sets_large_format_flag(self):
current_working_directory = os.getcwd()
file_path_npy = os.path.join(current_working_directory, "__file.npy")
numpy.save(file_path_npy, numpy.zeros((4, 4, 4)))
handler = ImportExportManager.NumPyImportExportHandler(
"numpy-io-handler", "npy", ["npy"])
try:
data_items = handler.read_data_items(None, "npy", file_path_npy)
self.assertEqual(len(data_items), 1)
data_item = data_items[0]
self.assertTrue(data_item.large_format)
finally:
os.remove(file_path_npy)
def test_npy_write_to_then_read_from_temp_file(self):
document_model = DocumentModel.DocumentModel()
with contextlib.closing(document_model):
current_working_directory = os.getcwd()
file_path_npy = os.path.join(current_working_directory, "__file.npy")
file_path_json = os.path.join(current_working_directory, "__file.json")
numpy.save(file_path_npy, numpy.zeros((16, 16)))
with open(file_path_json, "w") as f:
json.dump({"version": 1}, f)
handler = ImportExportManager.NumPyImportExportHandler("numpy-io-handler", "npy", ["npy"])
try:
data_items = handler.read_data_items(None, "npy", file_path_npy)
self.assertEqual(len(data_items), 1)
data_item = data_items[0]
finally:
os.remove(file_path_npy)
os.remove(file_path_json)
def test_npy_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_npy = os.path.join(current_working_directory,
"__file.npy")
file_path_json = os.path.join(current_working_directory,
"__file.json")
numpy.save(file_path_npy, numpy.zeros((16, 16)))
with open(file_path_json, "w") as f:
json.dump({"version": 1}, f)
handler = ImportExportManager.NumPyImportExportHandler(
"numpy-io-handler", "npy", ["npy"])
try:
data_items = handler.read_data_items(
"npy", pathlib.Path(file_path_npy))
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_npy)
os.remove(file_path_json)
