put class on parent scope

2025-03-21 15:03:12 -03:00
parent 85deafc08f
commit d21144e045
8 changed files with 42 additions and 34 deletions
--- a/layercake/tests/test_dynamodb.py
+++ b/layercake/tests/test_dynamodb.py
@@ -10,6 +10,7 @@ from layercake.dynamodb import (
    DynamoDBCollection,
    DynamoDBPersistenceLayer,
    KeyPair,
+    MissingError,
    PartitionKey,
    TransactItems,
    serialize,
@@ -103,7 +104,7 @@ def test_collection_get_item(
        'update_date': '2023-11-09T12:13:04.308986-03:00',
    }

-    with pytest.raises(DynamoDBCollection.MissingError):
+    with pytest.raises(MissingError):
        collect.get_item(key=KeyPair('5OxmMjL-ujoR5IMGegQz', 'notfound'))