added l0 and linf for count

pydp/algorithms/laplacian/_count.py

@@ -1,11 +1,24 @@
 from .._algorithm import MetaAlgorithm
 
+from typing import Union
+
 
 class Count(MetaAlgorithm):
     """
     Count Explanation
     TODO
     """
 
-    def __init__(self, epsilon: float = 1.0, dtype: str = "int"):
-        super().__init__(epsilon=epsilon, dtype=dtype)
+    def __init__(
+        self,
+        epsilon: float = 1.0,
+        l0_sensitivity: int = 1,
+        linf_sensitivity: int = 1,
+        dtype: str = "int",
+    ):
+        super().__init__(
+            epsilon=epsilon,
+            l0_sensitivity=l0_sensitivity,
+            linf_sensitivity=linf_sensitivity,
+            dtype=dtype,
+        )

tests/algorithms/test_count.py

@@ -67,6 +67,21 @@ def test_count_datatypes(self):
         assert isinstance(res, int)
 
 
+@pytest.mark.parametrize("dtype_in", ["int", "float"])
+class TestCount:
+    def test_basic(self, dtype_in):
+        n = 100
+        c = [1 for _ in range(100)]
+        count = Count(epsilon=1, dtype=dtype_in)
+        assert n - 10 < count.quick_result(c) < n + 10
+
+    def test_l0_linf(self, dtype_in):
+        n = 100
+        c = [1 for _ in range(100)]
+        count = Count(epsilon=1, l0_sensitivity=1, linf_sensitivity=1, dtype=dtype_in)
+        assert n - 10 < count.quick_result(c) < n + 10
+
+
 # TODO: port the following tests
 #
 # TEST(CountTest, MergeTest)