HUSARIA

code/conftest.py

@@ -0,0 +1,51 @@
+import pandas as pd
+import pytest
+
+from .project import sieve_of_eratosthenes
+
+
+def pytest_addoption(parser):
+    parser.addoption(
+        "--primes",
+        action="store",
+        default='100',
+        help="Upper bound for primes to test."
+    )
+    parser.addoption(
+        "--composites",
+        action="store",
+        default='100',
+        help="Upper bound for composites to test."
+    )
+
+
+def pytest_configure(config):
+    pytest.primes = int(config.getoption("--primes"))
+    pytest.composites = int(config.getoption("--composites"))
+
+
+@pytest.fixture
+def primes_obtained():
+    return set(sieve_of_eratosthenes(pytest.primes + 1))
+
+
+@pytest.fixture
+def composites_obtained():
+    return set(sieve_of_eratosthenes(pytest.composites + 1))
+
+def pytest_generate_tests(metafunc):
+    df_prime = pd.read_csv("P-100000.csv", header=None)
+    df_prime = df_prime[df_prime.iloc[:, 1] <  pytest.primes]
+    primes = set(df_prime.iloc[:, 1])
+    df_composite = pd.read_csv("C-100000.csv", header=None, delimiter="=")
+    # interpret first oclumns as numbers
+    df_composite = df_composite[df_composite.iloc[:, 0] < pytest.composites]
+    composites = set(df_composite.iloc[:, 0])
+    if metafunc.function.__name__ == 'test_positives':
+        metafunc.parametrize('prime', primes)
+        metafunc.parametrize('primes_obtained', [sieve_of_eratosthenes(max(primes) + 1),])
+
+    if metafunc.function.__name__ == 'test_negatives':
+        metafunc.parametrize('composite', composites)
+        
+        metafunc.parametrize('primes_obtained', [sieve_of_eratosthenes(max(composites) + 1),])

code/project.py

@@ -1,5 +1,4 @@
 '''Python program to print all Primes Smaller than or equal to N using Sieve of Eratosthenes'''
-import time
 import math
   
 def sieve_of_eratosthenes(num: int) -> list[int]:

code/test_main.py

@@ -0,0 +1,8 @@
+import pandas as pd
+
+def test_positives(prime, primes_obtained):
+    assert prime in primes_obtained
+
+
+def test_negatives(composite, primes_obtained):
+    assert composite not in primes_obtained