14 lines
534 B
Python
14 lines
534 B
Python
from src.portfolio.runner import rebalance_allocations
|
|
from src.portfolio.ledger import PortfolioLedger
|
|
|
|
|
|
def test_rebalance_resizes_to_total(tmp_path):
|
|
L = PortfolioLedger("PX", total_capital=1000.0, data_dir=tmp_path)
|
|
|
|
class FakeWorker:
|
|
def __init__(self, cap): self.capital = cap
|
|
workers = {"a": FakeWorker(700.0), "b": FakeWorker(500.0)}
|
|
rebalance_allocations(L, workers, {"a": 0.5, "b": 0.5})
|
|
assert L.total_capital == 1200.0
|
|
assert workers["a"].capital == 600.0 and workers["b"].capital == 600.0
|