28 lines
836 B
Python
28 lines
836 B
Python
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")
|