Ensure audb.Repository is hashable

audeering · Dec 15, 2024 · 2902074 · 2902074
1 parent d7b87e8
commit 2902074
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/audb/core/repository.py b/audb/core/repository.py
@@ -71,6 +71,15 @@ def __eq__(self, other) -> bool:
         """
         return str(self) == str(other)
 
+    def __hash__(self):
+        """Hash of repository.
+
+        Returns:
+            hash of repository
+
+        """
+        return hash(str(self))
+
     def __repr__(self):  # noqa: D105
         return (
             f"Repository("

diff --git a/tests/test_repository.py b/tests/test_repository.py
@@ -43,6 +43,17 @@ def test_repository_eq(repository1, repository2, expected):
     assert (repository1 == repository2) == expected
 
 
+@pytest.mark.parametrize(
+    "repository",
+    [audb.Repository("repo", "host", "file-system")],
+)
+def test_repository_hash(repository):
+    """Test Repository object is hashable."""
+    assert isinstance(hash(repository), int)
+    # set needs `__hash__` to exist to work
+    assert set([repository]) == set([str(repository)])
+
+
 @pytest.mark.parametrize(
     "backend, host, repo, expected",
     [