tests/test_db_layer.py

import pytest

from data_access import get_unified_sources


@pytest.mark.asyncio
async def test_get_unified_sources():
    results, stats = await get_unified_sources(2, 2, 1, 1)
    assert results is not None
    assert stats is not None

    # Check number of rows in results list
    assert len(results) > 4, "Results should contain at least one row"

    # Check number of rows in stats DataFrame
    assert stats.shape[0] > 0, "Stats DataFrame should contain at least one row"

    # You can also check specific stats columns
    assert "overlap_count" in stats.columns, "Stats should contain overlap_count"