mirror of
https://github.com/kuhyx/WUT_Computer_Science.git
synced 2026-07-04 17:43:12 +02:00
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
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"))
|
|
|
|
|
|
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_composites':
|
|
metafunc.parametrize('composite', composites)
|
|
metafunc.parametrize('primes_obtained', [sieve_of_eratosthenes(max(composites) + 1),])
|