from ..utils_dataset.px_chol import CLASS_NAME should work for an import like this. Adding .. will use the parent directory of test.py for the import.
from ..utils_dataset.px_chol import CLASS_NAME
..