Memory leak

Author: dgoeries

src/downsample/tests/test_lttb_memory.py src/downsample/tests/test_memory_leak.pysrc/downsample/tests/test_lttb_memory.py renamed to src/downsample/tests/test_memory_leak.py

@@ -2,28 +2,44 @@
 
 import numpy as np
 
-from downsample import largest_triangle_three_buckets
+from downsample import lttb, ltob, ltd
 
+import pytest
 
-def test_memory_leak():
+
+@pytest.mark.parametrize("func", [lttb, ltob])
+def test_memory_leak(func):
+    """
+    Test memory leak for different LTTB functions.
+
+    Args:
+        func (callable): The function to test.
+    """
     tracemalloc.start()
 
+    # Test parameters (shared for all functions)
     size = 1_000_000
+    threshold = 100
+    iterations = 1_000
+
+    # Generate test data
     x = np.linspace(0, 10, size)
     y = np.sin(x)
-    threshold = 100
 
+    # Snapshot before function execution
     before_snapshot = tracemalloc.take_snapshot()
 
-    for _ in range(1_000):
-        result = largest_triangle_three_buckets(x, y, threshold)
+    for _ in range(iterations):
+        result = func(x, y, threshold)
         del result
         import gc
         gc.collect()
 
+    # Snapshot after function execution
     after_snapshot = tracemalloc.take_snapshot()
     tracemalloc.stop()
 
+    # Calculate memory usage
     before_size = sum(
         stat.size for stat in before_snapshot.statistics("filename"))
     after_size = sum(