Implemented __eq__ function for Kovats and CubicSpline via ComputationMethod class

CHANGELOG.md

@@ -6,6 +6,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 ### Added
+- Added `__eq__` to `ComputationMethod` class and subclasses [#52](https://github.com/RECETOX/RIAssigner/pull/52)
 ### Changed
 ### Removed
 ## [0.1.0] - 2021-07-12

RIAssigner/compute/ComputationMethod.py

@@ -18,3 +18,6 @@ def _check_data_args(self, query, reference):
         """
         assert query is not None, "Query data is 'None'."
         assert reference is not None, "Reference data is 'None'."
+
+    def __eq__(self, o: object) -> bool:
+        return isinstance(o, type(self))

RIAssigner/compute/__init__.py

@@ -1,11 +1,13 @@
 import logging
 
+from .ComputationMethod import ComputationMethod
 from .CubicSpline import CubicSpline
 from .Kovats import Kovats
 
 logging.getLogger(__name__).addHandler(logging.NullHandler())
 
 __all__ = [
+    "ComputationMethod",
     "CubicSpline",
     "Kovats",
 ]

tests/test_compute_ComputationMethod.py

@@ -0,0 +1,24 @@
+import pytest
+
+from RIAssigner.compute import ComputationMethod, CubicSpline, Kovats
+
+
+@pytest.mark.parametrize('this, other, expected', [
+    [CubicSpline(), CubicSpline(), True],
+    [Kovats(), Kovats(), True],
+    [CubicSpline(), Kovats(), False],
+    [Kovats(), object(), False],
+    [CubicSpline(), object(), False],
+])
+def test_equal(this, other, expected):
+    actual = (this == other)
+    assert actual == expected
+
+
+def test_abc():
+    with pytest.raises(TypeError) as exception:
+        method = ComputationMethod()
+
+    message = exception.value.args[0]
+    assert exception.typename == "TypeError"
+    assert str(message).startswith("Can't instantiate abstract class ComputationMethod with abstract methods")