import polars as pl
from transformators.transformator import (
    transform_prices,
    transform_weather,
)

def test_transform_prices():
    df = pl.DataFrame({
        "timestamp": [1700000000000],
        "value": [50.0]
    })
    transformed = transform_prices(df)
    assert transformed.shape == (1, 2)
    assert transformed["timestamp"].dtype == pl.Datetime("ms", time_zone="UTC")

def test_transform_weather():
    df = pl.DataFrame({
        "timestamp": ["2023-11-14T22:13:20+00:00"],
        "temperature": [10.5],
        "wind_speed": [5.0],
        "solar": [0.0],
        "sunshine": [0.0],
        "cloud_cover": [0.0],
        "precipitation": [0.0]
    })
    transformed = transform_weather(df)
    assert transformed.shape == (1, 7)
    assert transformed["timestamp"].dtype == pl.Datetime("ms", time_zone="UTC")