diff --git a/tests/conftest.py b/tests/conftest.py index ab921dd..4e741cd 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -10,6 +10,8 @@ import pytest +from too.mock import create_mock_too_catalogue + DBNAME: str = "sdss5db_too_test" @@ -25,3 +27,10 @@ def sdss5db(): yield sdss5db.database.close() + + +@pytest.fixture(scope="session") +def too_mock(): + sample = create_mock_too_catalogue() + + yield sample diff --git a/tests/test_mock.py b/tests/test_mock.py new file mode 100644 index 0000000..9e2863b --- /dev/null +++ b/tests/test_mock.py @@ -0,0 +1,16 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- +# +# @Author: José Sánchez-Gallego (gallegoj@uw.edu) +# @Date: 2024-02-11 +# @Filename: test_mock.py +# @License: BSD 3-clause (http://www.opensource.org/licenses/BSD-3-Clause) + +from __future__ import annotations + +import polars + + +def test_mock_sample(too_mock: polars.DataFrame): + assert isinstance(too_mock, polars.DataFrame) + assert too_mock.height > 500000