counterweight/data-xexe
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

pairs are usable

Browse source
This commit is contained in:
Pablo Martin 2025-05-26 16:14:51 +02:00
parent 2432b7d6dc
commit 568e27adbe
3 changed files with 67 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

37
tests/tests_unit/test_utils.py
View file

@ -3,7 +3,12 @@ import datetime
import pytest
from money.currency import Currency
from xexe.utils import DateRange, generate_currency_and_dates_combinations
from xexe.currency_pair import CurrencyPair
from xexe.utils import (
DateRange,
generate_currency_and_dates_combinations,
generate_pairs_and_dates_combinations,
)
def test_date_range_breaks_with_reversed_dates():
@ -30,7 +35,7 @@ def test_date_range_generates_proper_dates_when_itered():
assert len(dates) == 3
def generate_currency_and_dates_combinations_outputs_correctly():
def test_generate_currency_and_dates_combinations_outputs_correctly():
date_range = DateRange(
start_date=datetime.date(year=2024, month=1, day=1),
@ -44,6 +49,28 @@ def generate_currency_and_dates_combinations_outputs_correctly():
)
assert len(combinations) == 9
assert len({date for date in combinations["date"]}) == 3
assert len({currency for currency in combinations["from_currency"]}) == 3
assert len({currency for currency in combinations["to_currency"]}) == 3
assert len({combination["date"] for combination in combinations}) == 3
assert len({combination["from_currency"] for combination in combinations}) == 2
assert len({combination["to_currency"] for combination in combinations}) == 2
def test_generate_pair_and_dates_combinations_outputs_correctly():
date_range = DateRange(
start_date=datetime.date(year=2024, month=1, day=1),
end_date=datetime.date(year=2024, month=1, day=3),
)
pairs = {
CurrencyPair(from_currency="USD", to_currency="EUR"),
CurrencyPair(from_currency="USD", to_currency="GBP"),
CurrencyPair(from_currency="EUR", to_currency="GBP"),
}
combinations = generate_pairs_and_dates_combinations(
pairs=pairs, date_range=date_range
)
assert len(combinations) == 9
assert len({combination["date"] for combination in combinations}) == 3
assert len({combination["from_currency"] for combination in combinations}) == 2
assert len({combination["to_currency"] for combination in combinations}) == 2