import pytest
import sys

# Ensure the local src/ is on PYTHONPATH for the test in this bare repo layout
sys.path.insert(0, "src")

from cosmosmesh_privacy_preserving_federated_ import add


def test_add_basic():
    assert add(2, 3) == 5