mirror of
https://github.com/bitcoinresearchkit/brk.git
synced 2026-04-24 14:49:58 -07:00
global: snapshot
This commit is contained in:
nym21
@@ -1,18 +1,20 @@
|
||||
# Tests for MetricData helper methods including polars/pandas conversion
|
||||
# Tests for MetricData and DateMetricData helper methods including polars/pandas conversion
|
||||
# Run: uv run pytest tests/test_metric_data.py -v
|
||||
|
||||
from datetime import date, datetime, timezone
|
||||
from datetime import date, datetime, timezone, timedelta
|
||||
|
||||
import pytest
|
||||
|
||||
from brk_client import MetricData
|
||||
from brk_client import MetricData, DateMetricData
|
||||
|
||||
|
||||
# ============ Fixtures ============
|
||||
|
||||
|
||||
# Test data fixtures
|
||||
@pytest.fixture
|
||||
def date_based_metric():
|
||||
"""MetricData with day1 (date-based)."""
|
||||
return MetricData(
|
||||
def day1_metric():
|
||||
"""DateMetricData with day1 (date-based, daily)."""
|
||||
return DateMetricData(
|
||||
version=1,
|
||||
index="day1",
|
||||
total=100,
|
||||
@@ -24,7 +26,7 @@ def date_based_metric():
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def height_based_metric():
|
||||
def height_metric():
|
||||
"""MetricData with height (non-date-based)."""
|
||||
return MetricData(
|
||||
version=1,
|
||||
@@ -38,9 +40,9 @@ def height_based_metric():
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def month_based_metric():
|
||||
"""MetricData with month1."""
|
||||
return MetricData(
|
||||
def month1_metric():
|
||||
"""DateMetricData with month1."""
|
||||
return DateMetricData(
|
||||
version=1,
|
||||
index="month1",
|
||||
total=200,
|
||||
@@ -51,103 +53,139 @@ def month_based_metric():
|
||||
)
|
||||
|
||||
|
||||
# ============ is_date_based tests ============
|
||||
@pytest.fixture
|
||||
def hour1_metric():
|
||||
"""DateMetricData with hour1 (sub-daily)."""
|
||||
return DateMetricData(
|
||||
version=1,
|
||||
index="hour1",
|
||||
total=200000,
|
||||
start=0,
|
||||
end=3,
|
||||
stamp="2024-01-01T00:00:00Z",
|
||||
data=[10.0, 20.0, 30.0],
|
||||
)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def minute5_metric():
|
||||
"""DateMetricData with minute5 (sub-daily)."""
|
||||
return DateMetricData(
|
||||
version=1,
|
||||
index="minute5",
|
||||
total=500000,
|
||||
start=0,
|
||||
end=3,
|
||||
stamp="2024-01-01T00:00:00Z",
|
||||
data=[1, 2, 3],
|
||||
)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def week1_metric():
|
||||
"""DateMetricData with week1."""
|
||||
return DateMetricData(
|
||||
version=1,
|
||||
index="week1",
|
||||
total=800,
|
||||
start=0,
|
||||
end=3,
|
||||
stamp="2024-01-01T00:00:00Z",
|
||||
data=[5, 10, 15],
|
||||
)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def year1_metric():
|
||||
"""DateMetricData with year1."""
|
||||
return DateMetricData(
|
||||
version=1,
|
||||
index="year1",
|
||||
total=20,
|
||||
start=0,
|
||||
end=3,
|
||||
stamp="2024-01-01T00:00:00Z",
|
||||
data=[100, 200, 300],
|
||||
)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def day3_metric():
|
||||
"""DateMetricData with day3."""
|
||||
return DateMetricData(
|
||||
version=1,
|
||||
index="day3",
|
||||
total=2000,
|
||||
start=0,
|
||||
end=3,
|
||||
stamp="2024-01-01T00:00:00Z",
|
||||
data=[7, 8, 9],
|
||||
)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def empty_metric():
|
||||
"""MetricData with empty data."""
|
||||
return MetricData(
|
||||
version=1,
|
||||
index="day1",
|
||||
total=100,
|
||||
start=5,
|
||||
end=5,
|
||||
stamp="2024-01-01T00:00:00Z",
|
||||
data=[],
|
||||
)
|
||||
|
||||
|
||||
# ============ is_date_based ============
|
||||
|
||||
|
||||
class TestIsDateBased:
|
||||
"""Test the is_date_based property."""
|
||||
def test_day1(self, day1_metric):
|
||||
assert day1_metric.is_date_based is True
|
||||
|
||||
def test_day1_is_date_based(self, date_based_metric):
|
||||
assert date_based_metric.is_date_based is True
|
||||
def test_month1(self, month1_metric):
|
||||
assert month1_metric.is_date_based is True
|
||||
|
||||
def test_height_is_not_date_based(self, height_based_metric):
|
||||
assert height_based_metric.is_date_based is False
|
||||
def test_hour1(self, hour1_metric):
|
||||
assert hour1_metric.is_date_based is True
|
||||
|
||||
def test_month1_is_date_based(self, month_based_metric):
|
||||
assert month_based_metric.is_date_based is True
|
||||
def test_minute5(self, minute5_metric):
|
||||
assert minute5_metric.is_date_based is True
|
||||
|
||||
def test_week1(self, week1_metric):
|
||||
assert week1_metric.is_date_based is True
|
||||
|
||||
def test_year1(self, year1_metric):
|
||||
assert year1_metric.is_date_based is True
|
||||
|
||||
def test_day3(self, day3_metric):
|
||||
assert day3_metric.is_date_based is True
|
||||
|
||||
def test_height(self, height_metric):
|
||||
assert height_metric.is_date_based is False
|
||||
|
||||
|
||||
# ============ Date conversion tests ============
|
||||
# ============ MetricData (int-indexed) ============
|
||||
|
||||
|
||||
class TestIndexToDate:
|
||||
"""Test the _index_to_date function via MetricData.dates()."""
|
||||
class TestMetricData:
|
||||
def test_keys(self, height_metric):
|
||||
assert height_metric.keys() == [800000, 800001, 800002, 800003, 800004]
|
||||
|
||||
def test_day1_zero(self, date_based_metric):
|
||||
"""day1 0 is genesis: Jan 3, 2009."""
|
||||
dates = date_based_metric.dates()
|
||||
assert dates[0] == date(2009, 1, 3)
|
||||
|
||||
def test_day1_one(self, date_based_metric):
|
||||
"""day1 1 is Jan 9, 2009 (6 day gap after genesis)."""
|
||||
dates = date_based_metric.dates()
|
||||
assert dates[1] == date(2009, 1, 9)
|
||||
|
||||
def test_day1_two(self, date_based_metric):
|
||||
"""day1 2 is Jan 10, 2009."""
|
||||
dates = date_based_metric.dates()
|
||||
assert dates[2] == date(2009, 1, 10)
|
||||
|
||||
def test_month1_dates(self, month_based_metric):
|
||||
"""month1 returns correct dates."""
|
||||
dates = month_based_metric.dates()
|
||||
assert dates[0] == date(2009, 1, 1)
|
||||
assert dates[1] == date(2009, 2, 1)
|
||||
assert dates[2] == date(2009, 3, 1)
|
||||
|
||||
|
||||
# ============ Smart keys/items/to_dict/iter tests ============
|
||||
|
||||
|
||||
class TestSmartHelpers:
|
||||
"""Test smart MetricData helpers that auto-detect date vs numeric keys."""
|
||||
|
||||
def test_keys_date_based(self, date_based_metric):
|
||||
"""keys() returns dates for date-based metric."""
|
||||
keys = date_based_metric.keys()
|
||||
assert len(keys) == 5
|
||||
assert keys[0] == date(2009, 1, 3) # genesis
|
||||
assert keys[1] == date(2009, 1, 9) # day1 1
|
||||
|
||||
def test_keys_height_based(self, height_based_metric):
|
||||
"""keys() returns index numbers for non-date-based metric."""
|
||||
keys = height_based_metric.keys()
|
||||
assert keys == [800000, 800001, 800002, 800003, 800004]
|
||||
|
||||
def test_items_date_based(self, date_based_metric):
|
||||
"""items() returns (date, value) pairs for date-based metric."""
|
||||
items = date_based_metric.items()
|
||||
assert items[0] == (date(2009, 1, 3), 100)
|
||||
assert items[1] == (date(2009, 1, 9), 200)
|
||||
|
||||
def test_items_height_based(self, height_based_metric):
|
||||
"""items() returns (index, value) pairs for height-based metric."""
|
||||
items = height_based_metric.items()
|
||||
def test_items(self, height_metric):
|
||||
items = height_metric.items()
|
||||
assert items[0] == (800000, 1.5)
|
||||
assert items[4] == (800004, 5.5)
|
||||
assert items[-1] == (800004, 5.5)
|
||||
|
||||
def test_to_dict_date_based(self, date_based_metric):
|
||||
"""to_dict() returns {date: value} for date-based metric."""
|
||||
d = date_based_metric.to_dict()
|
||||
assert d[date(2009, 1, 3)] == 100
|
||||
assert d[date(2009, 1, 9)] == 200
|
||||
|
||||
def test_to_dict_height_based(self, height_based_metric):
|
||||
"""to_dict() returns {index: value} for height-based metric."""
|
||||
d = height_based_metric.to_dict()
|
||||
def test_to_dict(self, height_metric):
|
||||
d = height_metric.to_dict()
|
||||
assert d[800000] == 1.5
|
||||
assert d[800004] == 5.5
|
||||
assert len(d) == 5
|
||||
|
||||
def test_iter_date_based(self, date_based_metric):
|
||||
"""Default iteration yields (date, value) for date-based metric."""
|
||||
result = list(date_based_metric)
|
||||
assert result[0] == (date(2009, 1, 3), 100)
|
||||
assert result[1] == (date(2009, 1, 9), 200)
|
||||
assert len(result) == 5
|
||||
|
||||
def test_iter_height_based(self, height_based_metric):
|
||||
"""Default iteration yields (index, value) for height-based metric."""
|
||||
result = list(height_based_metric)
|
||||
def test_iter(self, height_metric):
|
||||
result = list(height_metric)
|
||||
assert result == [
|
||||
(800000, 1.5),
|
||||
(800001, 2.5),
|
||||
@@ -156,130 +194,334 @@ class TestSmartHelpers:
|
||||
(800004, 5.5),
|
||||
]
|
||||
|
||||
def test_len(self, height_metric):
|
||||
assert len(height_metric) == 5
|
||||
|
||||
# ============ Explicit indexes/dates tests ============
|
||||
def test_indexes(self, height_metric):
|
||||
assert height_metric.indexes() == [800000, 800001, 800002, 800003, 800004]
|
||||
|
||||
def test_empty_data(self, empty_metric):
|
||||
assert len(empty_metric) == 0
|
||||
assert empty_metric.keys() == []
|
||||
assert empty_metric.items() == []
|
||||
assert empty_metric.to_dict() == {}
|
||||
assert list(empty_metric) == []
|
||||
assert empty_metric.indexes() == []
|
||||
|
||||
|
||||
class TestExplicitAccessors:
|
||||
"""Test explicit indexes() and dates() methods."""
|
||||
|
||||
def test_indexes_returns_range(self, date_based_metric):
|
||||
"""indexes() returns list of index values."""
|
||||
assert date_based_metric.indexes() == [0, 1, 2, 3, 4]
|
||||
|
||||
def test_indexes_with_offset(self, height_based_metric):
|
||||
"""indexes() respects start/end offsets."""
|
||||
assert height_based_metric.indexes() == [800000, 800001, 800002, 800003, 800004]
|
||||
|
||||
def test_dates_raises_for_non_date(self, height_based_metric):
|
||||
"""dates() raises for non-date-based index."""
|
||||
with pytest.raises(ValueError):
|
||||
height_based_metric.dates()
|
||||
# ============ DateMetricData inheritance ============
|
||||
|
||||
|
||||
# ============ Polars tests ============
|
||||
class TestDateMetricDataInheritance:
|
||||
def test_isinstance(self, day1_metric):
|
||||
assert isinstance(day1_metric, DateMetricData)
|
||||
assert isinstance(day1_metric, MetricData)
|
||||
|
||||
def test_int_keys_inherited(self, day1_metric):
|
||||
assert day1_metric.keys() == [0, 1, 2, 3, 4]
|
||||
|
||||
def test_int_items_inherited(self, day1_metric):
|
||||
items = day1_metric.items()
|
||||
assert items[0] == (0, 100)
|
||||
assert items[4] == (4, 500)
|
||||
|
||||
def test_int_iter_inherited(self, day1_metric):
|
||||
result = list(day1_metric)
|
||||
assert result[0] == (0, 100)
|
||||
|
||||
def test_len_inherited(self, day1_metric):
|
||||
assert len(day1_metric) == 5
|
||||
|
||||
def test_indexes_inherited(self, day1_metric):
|
||||
assert day1_metric.indexes() == [0, 1, 2, 3, 4]
|
||||
|
||||
def test_to_dict_inherited(self, day1_metric):
|
||||
d = day1_metric.to_dict()
|
||||
assert d[0] == 100
|
||||
assert isinstance(list(d.keys())[0], int)
|
||||
|
||||
|
||||
# ============ _index_to_date conversions ============
|
||||
|
||||
|
||||
class TestIndexToDate:
|
||||
"""Test date conversion for all index types."""
|
||||
|
||||
def test_day1_genesis(self, day1_metric):
|
||||
"""Day1 index 0 = 2009-01-03 (genesis)."""
|
||||
dates = day1_metric.dates()
|
||||
assert dates[0] == date(2009, 1, 3)
|
||||
|
||||
def test_day1_index_one(self, day1_metric):
|
||||
"""Day1 index 1 = 2009-01-09 (6-day gap after genesis)."""
|
||||
dates = day1_metric.dates()
|
||||
assert dates[1] == date(2009, 1, 9)
|
||||
|
||||
def test_day1_consecutive(self, day1_metric):
|
||||
"""Day1 indexes 2+ are consecutive days after index 1."""
|
||||
dates = day1_metric.dates()
|
||||
assert dates[2] == date(2009, 1, 10)
|
||||
assert dates[3] == date(2009, 1, 11)
|
||||
assert dates[4] == date(2009, 1, 12)
|
||||
|
||||
def test_day1_returns_date_type(self, day1_metric):
|
||||
dates = day1_metric.dates()
|
||||
assert type(dates[0]) is date
|
||||
|
||||
def test_month1(self, month1_metric):
|
||||
dates = month1_metric.dates()
|
||||
assert dates[0] == date(2009, 1, 1)
|
||||
assert dates[1] == date(2009, 2, 1)
|
||||
assert dates[2] == date(2009, 3, 1)
|
||||
assert type(dates[0]) is date
|
||||
|
||||
def test_week1(self, week1_metric):
|
||||
dates = week1_metric.dates()
|
||||
assert dates[0] == date(2009, 1, 3) # genesis
|
||||
assert dates[1] == date(2009, 1, 10) # +7 days
|
||||
assert dates[2] == date(2009, 1, 17) # +14 days
|
||||
assert type(dates[0]) is date
|
||||
|
||||
def test_year1(self, year1_metric):
|
||||
dates = year1_metric.dates()
|
||||
assert dates[0] == date(2009, 1, 1)
|
||||
assert dates[1] == date(2010, 1, 1)
|
||||
assert dates[2] == date(2011, 1, 1)
|
||||
assert type(dates[0]) is date
|
||||
|
||||
def test_day3(self, day3_metric):
|
||||
dates = day3_metric.dates()
|
||||
assert dates[0] == date(2009, 1, 1) # epoch
|
||||
assert dates[1] == date(2009, 1, 4) # +3 days
|
||||
assert dates[2] == date(2009, 1, 7) # +6 days
|
||||
|
||||
def test_hour1_returns_datetime(self, hour1_metric):
|
||||
"""Sub-daily indexes return datetime, not date."""
|
||||
dates = hour1_metric.dates()
|
||||
assert isinstance(dates[0], datetime)
|
||||
# hour1 index 0 = epoch (2009-01-01 00:00:00 UTC)
|
||||
assert dates[0] == datetime(2009, 1, 1, 0, 0, 0, tzinfo=timezone.utc)
|
||||
assert dates[1] == datetime(2009, 1, 1, 1, 0, 0, tzinfo=timezone.utc)
|
||||
assert dates[2] == datetime(2009, 1, 1, 2, 0, 0, tzinfo=timezone.utc)
|
||||
|
||||
def test_minute5_returns_datetime(self, minute5_metric):
|
||||
dates = minute5_metric.dates()
|
||||
assert isinstance(dates[0], datetime)
|
||||
assert dates[0] == datetime(2009, 1, 1, 0, 0, 0, tzinfo=timezone.utc)
|
||||
assert dates[1] == datetime(2009, 1, 1, 0, 5, 0, tzinfo=timezone.utc)
|
||||
assert dates[2] == datetime(2009, 1, 1, 0, 10, 0, tzinfo=timezone.utc)
|
||||
|
||||
|
||||
# ============ _date_to_index conversions ============
|
||||
|
||||
|
||||
class TestDateToIndex:
|
||||
"""Test reverse date-to-index conversion."""
|
||||
|
||||
def test_day1_genesis(self):
|
||||
from brk_client import _date_to_index
|
||||
assert _date_to_index("day1", date(2009, 1, 3)) == 0
|
||||
|
||||
def test_day1_before_day_one(self):
|
||||
from brk_client import _date_to_index
|
||||
# Dates before day1 1 map to 0
|
||||
assert _date_to_index("day1", date(2009, 1, 5)) == 0
|
||||
|
||||
def test_day1_index_one(self):
|
||||
from brk_client import _date_to_index
|
||||
assert _date_to_index("day1", date(2009, 1, 9)) == 1
|
||||
|
||||
def test_day1_later(self):
|
||||
from brk_client import _date_to_index
|
||||
assert _date_to_index("day1", date(2009, 1, 10)) == 2
|
||||
|
||||
def test_month1(self):
|
||||
from brk_client import _date_to_index
|
||||
assert _date_to_index("month1", date(2009, 1, 1)) == 0
|
||||
assert _date_to_index("month1", date(2009, 2, 1)) == 1
|
||||
assert _date_to_index("month1", date(2010, 1, 1)) == 12
|
||||
|
||||
def test_year1(self):
|
||||
from brk_client import _date_to_index
|
||||
assert _date_to_index("year1", date(2009, 1, 1)) == 0
|
||||
assert _date_to_index("year1", date(2010, 6, 15)) == 1
|
||||
assert _date_to_index("year1", date(2020, 1, 1)) == 11
|
||||
|
||||
def test_week1(self):
|
||||
from brk_client import _date_to_index
|
||||
assert _date_to_index("week1", date(2009, 1, 3)) == 0
|
||||
assert _date_to_index("week1", date(2009, 1, 10)) == 1
|
||||
|
||||
def test_hour1_with_datetime(self):
|
||||
from brk_client import _date_to_index
|
||||
epoch = datetime(2009, 1, 1, tzinfo=timezone.utc)
|
||||
assert _date_to_index("hour1", epoch) == 0
|
||||
assert _date_to_index("hour1", epoch + timedelta(hours=1)) == 1
|
||||
assert _date_to_index("hour1", epoch + timedelta(hours=24)) == 24
|
||||
|
||||
def test_minute5_with_datetime(self):
|
||||
from brk_client import _date_to_index
|
||||
epoch = datetime(2009, 1, 1, tzinfo=timezone.utc)
|
||||
assert _date_to_index("minute5", epoch) == 0
|
||||
assert _date_to_index("minute5", epoch + timedelta(minutes=5)) == 1
|
||||
assert _date_to_index("minute5", epoch + timedelta(minutes=12)) == 2 # floor
|
||||
|
||||
def test_hour1_with_plain_date(self):
|
||||
"""Plain date is treated as midnight UTC for sub-daily."""
|
||||
from brk_client import _date_to_index
|
||||
# 2009-01-01 as date → midnight UTC → index 0
|
||||
assert _date_to_index("hour1", date(2009, 1, 1)) == 0
|
||||
# 2009-01-02 as date → midnight UTC → 24 hours later
|
||||
assert _date_to_index("hour1", date(2009, 1, 2)) == 24
|
||||
|
||||
def test_roundtrip_day1(self):
|
||||
"""date → index → date roundtrip for day1."""
|
||||
from brk_client import _date_to_index, _index_to_date
|
||||
for i in range(10):
|
||||
d = _index_to_date("day1", i)
|
||||
assert _date_to_index("day1", d) == i
|
||||
|
||||
def test_roundtrip_month1(self):
|
||||
from brk_client import _date_to_index, _index_to_date
|
||||
for i in range(24):
|
||||
d = _index_to_date("month1", i)
|
||||
assert _date_to_index("month1", d) == i
|
||||
|
||||
def test_roundtrip_hour1(self):
|
||||
from brk_client import _date_to_index, _index_to_date
|
||||
for i in range(48):
|
||||
d = _index_to_date("hour1", i)
|
||||
assert _date_to_index("hour1", d) == i
|
||||
|
||||
|
||||
# ============ DateMetricData date methods ============
|
||||
|
||||
|
||||
class TestDateMetricDataMethods:
|
||||
def test_date_items(self, day1_metric):
|
||||
items = day1_metric.date_items()
|
||||
assert items[0] == (date(2009, 1, 3), 100)
|
||||
assert items[1] == (date(2009, 1, 9), 200)
|
||||
assert len(items) == 5
|
||||
|
||||
def test_to_date_dict(self, day1_metric):
|
||||
d = day1_metric.to_date_dict()
|
||||
assert d[date(2009, 1, 3)] == 100
|
||||
assert d[date(2009, 1, 9)] == 200
|
||||
assert len(d) == 5
|
||||
# Keys should be date objects
|
||||
assert type(list(d.keys())[0]) is date
|
||||
|
||||
def test_date_items_sub_daily(self, hour1_metric):
|
||||
items = hour1_metric.date_items()
|
||||
assert isinstance(items[0][0], datetime)
|
||||
assert items[0] == (datetime(2009, 1, 1, 0, 0, 0, tzinfo=timezone.utc), 10.0)
|
||||
|
||||
def test_to_date_dict_sub_daily(self, hour1_metric):
|
||||
d = hour1_metric.to_date_dict()
|
||||
key = datetime(2009, 1, 1, 0, 0, 0, tzinfo=timezone.utc)
|
||||
assert d[key] == 10.0
|
||||
assert isinstance(list(d.keys())[0], datetime)
|
||||
|
||||
|
||||
# ============ Polars ============
|
||||
|
||||
|
||||
class TestPolarsConversion:
|
||||
"""Test MetricData.to_polars() conversion."""
|
||||
|
||||
@pytest.fixture(autouse=True)
|
||||
def check_polars(self):
|
||||
"""Skip if polars not installed."""
|
||||
pytest.importorskip("polars")
|
||||
|
||||
def test_to_polars_with_dates(self, date_based_metric):
|
||||
"""to_polars() includes date column for date-based index."""
|
||||
def test_metric_data_to_polars(self, height_metric):
|
||||
import polars as pl
|
||||
df = height_metric.to_polars()
|
||||
assert isinstance(df, pl.DataFrame)
|
||||
assert list(df.columns) == ["index", "value"]
|
||||
assert df["index"].to_list() == [800000, 800001, 800002, 800003, 800004]
|
||||
assert df["value"].to_list() == [1.5, 2.5, 3.5, 4.5, 5.5]
|
||||
|
||||
df = date_based_metric.to_polars()
|
||||
def test_date_metric_to_polars_with_dates(self, day1_metric):
|
||||
import polars as pl
|
||||
df = day1_metric.to_polars()
|
||||
assert isinstance(df, pl.DataFrame)
|
||||
assert "date" in df.columns
|
||||
assert "value" in df.columns
|
||||
assert "index" not in df.columns
|
||||
assert len(df) == 5
|
||||
assert df["value"].to_list() == [100, 200, 300, 400, 500]
|
||||
|
||||
def test_to_polars_without_dates(self, date_based_metric):
|
||||
"""to_polars(with_dates=False) uses index column."""
|
||||
def test_date_metric_to_polars_without_dates(self, day1_metric):
|
||||
import polars as pl
|
||||
|
||||
df = date_based_metric.to_polars(with_dates=False)
|
||||
df = day1_metric.to_polars(with_dates=False)
|
||||
assert "index" in df.columns
|
||||
assert "date" not in df.columns
|
||||
assert df["index"].to_list() == [0, 1, 2, 3, 4]
|
||||
|
||||
def test_to_polars_non_date_index(self, height_based_metric):
|
||||
"""to_polars() uses index column for non-date-based index."""
|
||||
import polars as pl
|
||||
|
||||
df = height_based_metric.to_polars()
|
||||
assert "index" in df.columns
|
||||
assert "date" not in df.columns
|
||||
assert df["index"].to_list() == [800000, 800001, 800002, 800003, 800004]
|
||||
assert df["value"].to_list() == [1.5, 2.5, 3.5, 4.5, 5.5]
|
||||
|
||||
def test_to_polars_month1(self, month_based_metric):
|
||||
"""to_polars() works with month1."""
|
||||
import polars as pl
|
||||
|
||||
df = month_based_metric.to_polars()
|
||||
def test_month1_to_polars(self, month1_metric):
|
||||
df = month1_metric.to_polars()
|
||||
assert "date" in df.columns
|
||||
assert len(df) == 3
|
||||
dates = df["date"].to_list()
|
||||
assert dates[0] == date(2009, 1, 1)
|
||||
assert dates[1] == date(2009, 2, 1)
|
||||
assert dates[2] == date(2009, 3, 1)
|
||||
|
||||
def test_sub_daily_to_polars(self, hour1_metric):
|
||||
df = hour1_metric.to_polars()
|
||||
assert "date" in df.columns
|
||||
assert len(df) == 3
|
||||
|
||||
# ============ Pandas tests ============
|
||||
def test_empty_to_polars(self, empty_metric):
|
||||
df = empty_metric.to_polars()
|
||||
assert len(df) == 0
|
||||
assert list(df.columns) == ["index", "value"]
|
||||
|
||||
|
||||
# ============ Pandas ============
|
||||
|
||||
|
||||
class TestPandasConversion:
|
||||
"""Test MetricData.to_pandas() conversion."""
|
||||
|
||||
@pytest.fixture(autouse=True)
|
||||
def check_pandas(self):
|
||||
"""Skip if pandas not installed."""
|
||||
pytest.importorskip("pandas")
|
||||
|
||||
def test_to_pandas_with_dates(self, date_based_metric):
|
||||
"""to_pandas() includes date column for date-based index."""
|
||||
def test_metric_data_to_pandas(self, height_metric):
|
||||
import pandas as pd
|
||||
df = height_metric.to_pandas()
|
||||
assert isinstance(df, pd.DataFrame)
|
||||
assert list(df.columns) == ["index", "value"]
|
||||
assert df["index"].tolist() == [800000, 800001, 800002, 800003, 800004]
|
||||
assert df["value"].tolist() == [1.5, 2.5, 3.5, 4.5, 5.5]
|
||||
|
||||
df = date_based_metric.to_pandas()
|
||||
def test_date_metric_to_pandas_with_dates(self, day1_metric):
|
||||
import pandas as pd
|
||||
df = day1_metric.to_pandas()
|
||||
assert isinstance(df, pd.DataFrame)
|
||||
assert "date" in df.columns
|
||||
assert "value" in df.columns
|
||||
assert len(df) == 5
|
||||
assert df["value"].tolist() == [100, 200, 300, 400, 500]
|
||||
|
||||
def test_to_pandas_without_dates(self, date_based_metric):
|
||||
"""to_pandas(with_dates=False) uses index column."""
|
||||
def test_date_metric_to_pandas_without_dates(self, day1_metric):
|
||||
import pandas as pd
|
||||
|
||||
df = date_based_metric.to_pandas(with_dates=False)
|
||||
df = day1_metric.to_pandas(with_dates=False)
|
||||
assert "index" in df.columns
|
||||
assert "date" not in df.columns
|
||||
assert df["index"].tolist() == [0, 1, 2, 3, 4]
|
||||
|
||||
def test_to_pandas_non_date_index(self, height_based_metric):
|
||||
"""to_pandas() uses index column for non-date-based index."""
|
||||
import pandas as pd
|
||||
|
||||
df = height_based_metric.to_pandas()
|
||||
assert "index" in df.columns
|
||||
assert "date" not in df.columns
|
||||
assert df["index"].tolist() == [800000, 800001, 800002, 800003, 800004]
|
||||
assert df["value"].tolist() == [1.5, 2.5, 3.5, 4.5, 5.5]
|
||||
|
||||
def test_to_pandas_month1(self, month_based_metric):
|
||||
"""to_pandas() works with month1."""
|
||||
import pandas as pd
|
||||
|
||||
df = month_based_metric.to_pandas()
|
||||
def test_month1_to_pandas(self, month1_metric):
|
||||
df = month1_metric.to_pandas()
|
||||
assert "date" in df.columns
|
||||
assert len(df) == 3
|
||||
dates = df["date"].tolist()
|
||||
assert dates[0] == date(2009, 1, 1)
|
||||
assert dates[1] == date(2009, 2, 1)
|
||||
assert dates[2] == date(2009, 3, 1)
|
||||
|
||||
def test_sub_daily_to_pandas(self, hour1_metric):
|
||||
df = hour1_metric.to_pandas()
|
||||
assert "date" in df.columns
|
||||
assert len(df) == 3
|
||||
|
||||
def test_empty_to_pandas(self, empty_metric):
|
||||
df = empty_metric.to_pandas()
|
||||
assert len(df) == 0
|
||||
assert list(df.columns) == ["index", "value"]
|
||||
|
||||
Reference in New Issue
Block a user