Mirror/brk
1
0
Fork 0
mirror of https://github.com/bitcoinresearchkit/brk.git synced 2026-04-24 14:49:58 -07:00
Code Issues Packages Projects Releases Wiki Activity

global: MASSIVE snapshot

Browse Source
This commit is contained in:
nym21
2026-02-23 17:22:12 +01:00
parent be0d749f9c
commit 3b7aa8242a
703 changed files with 29130 additions and 30779 deletions
Download Patch File Download Diff File Expand all files Collapse all files

287
packages/brk_client/tests/test_metric_data.py
View File

@@ -1,19 +1,20 @@
# Tests for MetricData helper methods including polars/pandas conversion
# Run: uv run pytest tests/test_metric_data.py -v
from datetime import date
from datetime import date, datetime, timezone
import pytest
from brk_client import MetricData, index_to_date, is_date_index
from brk_client import MetricData
# Test data fixtures
@pytest.fixture
def date_based_metric():
"""MetricData with dateindex (date-based)."""
"""MetricData with day1 (date-based)."""
return MetricData(
version=1,
index="dateindex",
index="day1",
total=100,
start=0,
end=5,
@@ -38,10 +39,10 @@ def height_based_metric():
@pytest.fixture
def month_based_metric():
"""MetricData with monthindex."""
"""MetricData with month1."""
return MetricData(
version=1,
index="monthindex",
index="month1",
total=200,
start=0,
end=3,
@@ -50,124 +51,117 @@ def month_based_metric():
)
# ============ is_date_based tests ============
class TestIsDateBased:
"""Test the is_date_based property."""
def test_day1_is_date_based(self, date_based_metric):
assert date_based_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_month1_is_date_based(self, month_based_metric):
assert month_based_metric.is_date_based is True
# ============ Date conversion tests ============
class TestIndexToDate:
"""Test the index_to_date function."""
"""Test the _index_to_date function via MetricData.dates()."""
def test_dateindex_zero(self):
"""DateIndex 0 is genesis: Jan 3, 2009."""
assert index_to_date("dateindex", 0) == date(2009, 1, 3)
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_dateindex_one(self):
"""DateIndex 1 is Jan 9, 2009 (6 day gap after genesis)."""
assert index_to_date("dateindex", 1) == date(2009, 1, 9)
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_dateindex_two(self):
"""DateIndex 2 is Jan 10, 2009."""
assert index_to_date("dateindex", 2) == date(2009, 1, 10)
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_weekindex_zero(self):
"""WeekIndex 0 is genesis: Jan 3, 2009."""
assert index_to_date("weekindex", 0) == date(2009, 1, 3)
def test_weekindex_one(self):
"""WeekIndex 1 is Jan 10, 2009 (one week after genesis)."""
assert index_to_date("weekindex", 1) == date(2009, 1, 10)
def test_monthindex_zero(self):
"""MonthIndex 0 is Jan 1, 2009."""
assert index_to_date("monthindex", 0) == date(2009, 1, 1)
def test_monthindex_one(self):
"""MonthIndex 1 is Feb 1, 2009."""
assert index_to_date("monthindex", 1) == date(2009, 2, 1)
def test_monthindex_twelve(self):
"""MonthIndex 12 is Jan 1, 2010."""
assert index_to_date("monthindex", 12) == date(2010, 1, 1)
def test_yearindex_zero(self):
"""YearIndex 0 is Jan 1, 2009."""
assert index_to_date("yearindex", 0) == date(2009, 1, 1)
def test_yearindex_one(self):
"""YearIndex 1 is Jan 1, 2010."""
assert index_to_date("yearindex", 1) == date(2010, 1, 1)
def test_quarterindex_zero(self):
"""QuarterIndex 0 is Q1 2009: Jan 1, 2009."""
assert index_to_date("quarterindex", 0) == date(2009, 1, 1)
def test_quarterindex_one(self):
"""QuarterIndex 1 is Q2 2009: Apr 1, 2009."""
assert index_to_date("quarterindex", 1) == date(2009, 4, 1)
def test_quarterindex_four(self):
"""QuarterIndex 4 is Q1 2010: Jan 1, 2010."""
assert index_to_date("quarterindex", 4) == date(2010, 1, 1)
def test_semesterindex_zero(self):
"""SemesterIndex 0 is H1 2009: Jan 1, 2009."""
assert index_to_date("semesterindex", 0) == date(2009, 1, 1)
def test_semesterindex_one(self):
"""SemesterIndex 1 is H2 2009: Jul 1, 2009."""
assert index_to_date("semesterindex", 1) == date(2009, 7, 1)
def test_semesterindex_two(self):
"""SemesterIndex 2 is H1 2010: Jan 1, 2010."""
assert index_to_date("semesterindex", 2) == date(2010, 1, 1)
def test_decadeindex_zero(self):
"""DecadeIndex 0 is 2009: Jan 1, 2009."""
assert index_to_date("decadeindex", 0) == date(2009, 1, 1)
def test_decadeindex_one(self):
"""DecadeIndex 1 is 2019: Jan 1, 2019."""
assert index_to_date("decadeindex", 1) == date(2019, 1, 1)
def test_invalid_index_raises(self):
"""Non-date-based index raises ValueError."""
with pytest.raises(ValueError):
index_to_date("height", 0)
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)
class TestIsDateIndex:
"""Test the is_date_index function."""
def test_dateindex_is_date_based(self):
assert is_date_index("dateindex") is True
def test_weekindex_is_date_based(self):
assert is_date_index("weekindex") is True
def test_monthindex_is_date_based(self):
assert is_date_index("monthindex") is True
def test_yearindex_is_date_based(self):
assert is_date_index("yearindex") is True
def test_quarterindex_is_date_based(self):
assert is_date_index("quarterindex") is True
def test_semesterindex_is_date_based(self):
assert is_date_index("semesterindex") is True
def test_decadeindex_is_date_based(self):
assert is_date_index("decadeindex") is True
def test_height_is_not_date_based(self):
assert is_date_index("height") is False
def test_txindex_is_not_date_based(self):
assert is_date_index("txindex") is False
# ============ Smart keys/items/to_dict/iter tests ============
# ============ MetricData helper method tests ============
class TestSmartHelpers:
"""Test smart MetricData helpers that auto-detect date vs numeric keys."""
class TestMetricDataHelpers:
"""Test MetricData helper methods."""
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()
assert items[0] == (800000, 1.5)
assert items[4] == (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()
assert d[800000] == 1.5
assert d[800004] == 5.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)
assert result == [
(800000, 1.5),
(800001, 2.5),
(800002, 3.5),
(800003, 4.5),
(800004, 5.5),
]
# ============ Explicit indexes/dates tests ============
class TestExplicitAccessors:
"""Test explicit indexes() and dates() methods."""
def test_indexes_returns_range(self, date_based_metric):
"""indexes() returns list of index values."""
@@ -177,64 +171,15 @@ class TestMetricDataHelpers:
"""indexes() respects start/end offsets."""
assert height_based_metric.indexes() == [800000, 800001, 800002, 800003, 800004]
def test_dates_for_dateindex(self, date_based_metric):
"""dates() returns correct dates for dateindex."""
dates = date_based_metric.dates()
assert dates[0] == date(2009, 1, 3) # dateindex 0 = genesis
assert dates[1] == date(2009, 1, 9) # dateindex 1 = day one
assert dates[2] == date(2009, 1, 10) # dateindex 2
def test_dates_for_monthindex(self, month_based_metric):
"""dates() returns correct dates for monthindex."""
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)
def test_to_index_dict(self, date_based_metric):
"""to_index_dict() returns {index: value} mapping."""
d = date_based_metric.to_index_dict()
assert d[0] == 100
assert d[1] == 200
assert d[4] == 500
def test_to_date_dict(self, date_based_metric):
"""to_date_dict() returns {date: value} mapping."""
d = date_based_metric.to_date_dict()
assert d[date(2009, 1, 3)] == 100 # genesis
assert d[date(2009, 1, 9)] == 200 # day one
def test_index_items(self, date_based_metric):
"""index_items() returns [(index, value), ...] pairs."""
items = date_based_metric.index_items()
assert items[0] == (0, 100)
assert items[4] == (4, 500)
def test_date_items(self, date_based_metric):
"""date_items() returns [(date, value), ...] pairs."""
items = date_based_metric.date_items()
assert items[0] == (date(2009, 1, 3), 100)
assert items[1] == (date(2009, 1, 9), 200)
def test_iter(self, date_based_metric):
"""iter() yields (index, value) pairs."""
result = list(date_based_metric.iter())
assert result == [(0, 100), (1, 200), (2, 300), (3, 400), (4, 500)]
def test_iter_dates(self, date_based_metric):
"""iter_dates() yields (date, value) pairs."""
result = list(date_based_metric.iter_dates())
assert result[0] == (date(2009, 1, 3), 100)
assert result[1] == (date(2009, 1, 9), 200)
def test_default_iteration(self, date_based_metric):
"""Default iteration uses iter()."""
result = list(date_based_metric)
assert result == [(0, 100), (1, 200), (2, 300), (3, 400), (4, 500)]
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()
# ============ Polars tests ============
class TestPolarsConversion:
"""Test MetricData.to_polars() conversion."""
@@ -273,14 +218,13 @@ class TestPolarsConversion:
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_monthindex(self, month_based_metric):
"""to_polars() works with monthindex."""
def test_to_polars_month1(self, month_based_metric):
"""to_polars() works with month1."""
import polars as pl
df = month_based_metric.to_polars()
assert "date" in df.columns
assert len(df) == 3
# Polars stores dates as date type
dates = df["date"].to_list()
assert dates[0] == date(2009, 1, 1)
assert dates[1] == date(2009, 2, 1)
@@ -289,6 +233,7 @@ class TestPolarsConversion:
# ============ Pandas tests ============
class TestPandasConversion:
"""Test MetricData.to_pandas() conversion."""
@@ -327,8 +272,8 @@ class TestPandasConversion:
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_monthindex(self, month_based_metric):
"""to_pandas() works with monthindex."""
def test_to_pandas_month1(self, month_based_metric):
"""to_pandas() works with month1."""
import pandas as pd
df = month_based_metric.to_pandas()