diff --git a/pyo3-object_store/src/error.rs b/pyo3-object_store/src/error.rs index 27149418..6d803660 100644 --- a/pyo3-object_store/src/error.rs +++ b/pyo3-object_store/src/error.rs @@ -1,7 +1,7 @@ //! Contains the [`PyObjectStoreError`], the error enum returned by all fallible functions in this //! crate. -use pyo3::exceptions::{PyFileNotFoundError, PyIOError, PyNotImplementedError, PyValueError}; +use pyo3::exceptions::{PyIOError, PyNotImplementedError, PyValueError}; use pyo3::prelude::*; use pyo3::{create_exception, DowncastError}; use thiserror::Error; @@ -113,7 +113,7 @@ impl From for PyErr { source: _, } => GenericError::new_err(format!("{err:#?}")), object_store::Error::NotFound { path: _, source: _ } => { - PyFileNotFoundError::new_err(format!("{err:#?}")) + NotFoundError::new_err(format!("{err:#?}")) } object_store::Error::InvalidPath { source: _ } => { InvalidPathError::new_err(format!("{err:#?}")) diff --git a/tests/test_delete.py b/tests/test_delete.py index d0ba0813..105c4f81 100644 --- a/tests/test_delete.py +++ b/tests/test_delete.py @@ -3,6 +3,7 @@ import pytest import obstore as obs +from obstore.exceptions import NotFoundError from obstore.store import LocalStore, MemoryStore @@ -50,7 +51,7 @@ def test_delete_one_local_fs(): obs.delete(store, "file3.txt") assert len(obs.list(store).collect()) == 0 - with pytest.raises(FileNotFoundError): + with pytest.raises(NotFoundError): obs.delete(store, "file1.txt") @@ -68,7 +69,7 @@ def test_delete_many_local_fs(): ["file1.txt", "file2.txt", "file3.txt"], ) - with pytest.raises(FileNotFoundError): + with pytest.raises(NotFoundError): obs.delete( store, ["file1.txt", "file2.txt", "file3.txt"],