import path

pyproject.toml

@@ -7,6 +7,7 @@ requires-python = ">=3.11"
 
 [tool.pytest.ini_options]
 testpaths = ["tests"]
+pythonpath = ["src"]
 
 [project.scripts]
 tea3-linear-bias = "tea3.tea3linearbias:main"

src/tea3/tea3.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 from dataclasses import dataclass
 from typing import Sequence
 
-from .constants import TEA3_LUT_A, TEA3_LUT_B, TEA3_SBOX
+from tea3.constants import TEA3_LUT_A, TEA3_LUT_B, TEA3_SBOX
 
 MASK32 = 0xFFFFFFFF
 MASK64 = 0xFFFFFFFFFFFFFFFF

src/tea3/tea3differentialbias.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 from dataclasses import dataclass
 from random import Random
 
-from .tea3 import Tea3
+from tea3.tea3 import Tea3
 
 
 def rand_key(rng: Random) -> list[int]:

src/tea3/tea3linearbias.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 from dataclasses import dataclass
 from random import Random
 
-from .tea3 import Tea3
+from tea3.tea3 import Tea3
 
 
 def rand_key(rng: Random) -> list[int]: