f-s: fix tests

src/aiida_pseudo/data/pseudo/pseudo.py

@@ -7,19 +7,28 @@
 import typing
 
 from aiida import orm, plugins
+from aiida.orm.nodes.caching import NodeCaching
 from aiida.common.constants import elements
 from aiida.common.exceptions import StoringNotAllowed
 from aiida.common.files import md5_from_filelike
 
 __all__ = ('PseudoPotentialData',)
 
+class PseudoPotentialDataCaching(NodeCaching):
+    """Class to define caching behavior of ``PseudoPotentialData`` nodes."""
+
+    def _get_objects_to_hash(self) -> list:
+        """Return a list of objects which should be included in the node hash."""
+        return [self._node.element, self._node.md5]
 
 class PseudoPotentialData(plugins.DataFactory('core.singlefile')):
     """Base class for data types representing pseudo potentials."""
 
     _key_element = 'element'
     _key_md5 = 'md5'
 
+    _CLS_NODE_CACHING = PseudoPotentialDataCaching
+
     @classmethod
     def get_or_create(cls, source: typing.Union[str, pathlib.Path, typing.BinaryIO], filename: str = None):
         """Get pseudopotenial data node from database with matching md5 checksum or create a new one if not existent.
@@ -184,7 +193,3 @@ def md5(self, value: str):
         """
         self.validate_md5(value)
         self.base.attributes.set(self._key_md5, value)
-
-    def _get_objects_to_hash(self) -> list:
-        """Return a list of objects which should be included in the node hash."""
-        return [self.element, self.md5]

tests/data/pseudo/test_pseudo.py

@@ -212,8 +212,5 @@ def test_get_or_create(get_pseudo_potential_data):
     assert different_class.uuid != original.uuid
 
     # check hash is from content and not from filename
-    print(original._get_objects_to_hash())
-    print([original.element, original.md5])
-    print(make_hash(original._get_objects_to_hash()))
     assert original._get_objects_to_hash() == [original.element, original.md5]
     assert original.get_hash() == make_hash([original.element, original.md5])