feat : unit tests for decorators and generators.

Author: manas-shinde

tests/test_cache.py

@@ -0,0 +1,20 @@
+import time
+from decorators.cache_decorator import cache_result
+
+
+@cache_result(ttl_seconds=2)
+def slow_square(x):
+    time.sleep(1)
+    return x * x
+
+
+def test_cache_result():
+    start = time.time()
+    assert slow_square(3) == 9
+    first_duration = time.time() - start
+
+    start = time.time()
+    assert slow_square(3) == 9  # From cache
+    second_duration = time.time() - start
+
+    assert second_duration < first_duration  # Cached

tests/test_fibonacci.py

@@ -0,0 +1,6 @@
+from generators.fibonacci import fibonacci
+
+def test_fibonacci():
+    fib = fibonacci()
+    results = [next(fib) for _ in range(6)]
+    assert results == [0, 1, 1, 2, 3, 5]

tests/test_file_chunker.py

@@ -0,0 +1,10 @@
+import tempfile
+from generators.file_chunker import file_chunker
+
+
+def test_file_chunker_reads_data():
+    with tempfile.NamedTemporaryFile(mode='wb', delete=False) as tf:
+        tf.write(b"abcdefghij" * 100)
+
+    chunks = list(file_chunker(tf.name, chunk_size=100))
+    assert sum(len(c) for c in chunks) == 1000

tests/test_logging.py

@@ -0,0 +1,11 @@
+from decorators.logging_decorator import log_execution
+
+
+@log_execution
+def add(a, b):
+    return a + b
+
+
+def test_log_execution():
+    assert add(2, 3) == 5
+    print("Success")

tests/test_retry.py

@@ -0,0 +1,19 @@
+import pytest
+from decorators.retry_decorator import retry_on_exception
+
+counter = {"calls": 0}
+
+
+@retry_on_exception(retries=3, delay=0)
+def flaky_function():
+    counter["calls"] += 1
+    if counter["calls"] < 3:
+        raise ValueError("Fail")
+    return "success"
+
+
+def test_retry_on_exception():
+    counter["calls"] = 0
+    result = flaky_function()
+    assert result == "success"
+    assert counter["calls"] == 3

tests/test_time.py

@@ -0,0 +1,10 @@
+from decorators.time_decorator import time_execution
+
+
+@time_execution
+def add(x, y):
+    return x + y
+
+
+def test_time_execution():
+    assert add(2, 5) == 7