"""Generate raw binary dataframe."""

from pathlib import Path

from psycopmlutils.synth_data_generator.synth_col_generators import (
    generate_data_columns,
)

if __name__ == "__main__":
    # Get project root directory
    project_root = Path(__file__).resolve().parents[3]

    column_specs = {
        "dw_ek_borger": {
            "column_type": "uniform_int",
            "min": 0,
            "max": 10_000,
        },
        "timestamp": {
            "column_type": "datetime_uniform",
            "min": -5 * 365,
            "max": 0 * 365,
        },
        "value": {"column_type": "uniform_int", "min": 0, "max": 1},
    }

    for i in (1, 2):
        df = generate_data_columns(
            predictors=column_specs,
            n_samples=10_000,
        )

        df.to_csv(
            project_root / "tests" / "test_data" / "raw" / f"synth_raw_binary_{i}.csv",
            index=False,
        )
