Colin Maudry
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
import datetime
|
|
import os
|
|
|
|
import polars as pl
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture(scope="session", autouse=True)
|
|
def test_data():
|
|
data = [
|
|
{
|
|
"uid": "1",
|
|
"acheteur_nom": "Acheteur 1",
|
|
"acheteur_id": "a1",
|
|
"titulaire_nom": "Titulaire 1",
|
|
"titulaire_id": "t1",
|
|
"montant": 10,
|
|
"dateNotification": datetime.date(2025, 1, 1),
|
|
"codeCPV": "71600000",
|
|
"donneesActuelles": True,
|
|
"acheteur_departement_code": "75",
|
|
"acheteur_departement_nom": "Paris",
|
|
"acheteur_commune_nom": "Paris",
|
|
"titulaire_departement_code": "35",
|
|
"titulaire_departement_nom": "Ille-et-Vilaine",
|
|
"titulaire_commune_nom": "Rennes",
|
|
"titulaire_typeIdentifiant": "SIRET",
|
|
"objet": "Objet test",
|
|
"dureeRestanteMois": 12,
|
|
"lieuExecution_code": "75001",
|
|
"sourceFile": "test.xml",
|
|
"sourceDataset": "test_dataset",
|
|
"datePublicationDonnees": datetime.date(2025, 1, 1),
|
|
}
|
|
]
|
|
path = "tests/test.parquet"
|
|
path = os.path.abspath(path)
|
|
print(f"Writing test data to: {path}") # <-- This will show you the real path
|
|
|
|
pl.DataFrame(data).write_parquet("tests/test.parquet")
|
|
yield path
|