26 lines
1.0 KiB
Python
26 lines
1.0 KiB
Python
from src.portfolio.runner import rebalance_allocations
|
|
from src.portfolio.ledger import PortfolioLedger
|
|
|
|
|
|
class FakeWorker:
|
|
def __init__(self, cap, in_position=False):
|
|
self.capital = cap
|
|
self.in_position = in_position
|
|
|
|
|
|
def test_rebalance_resizes_flat_workers(tmp_path):
|
|
L = PortfolioLedger("PX", total_capital=1000.0, data_dir=tmp_path)
|
|
workers = {"a": FakeWorker(700.0), "b": FakeWorker(500.0)} # equity 1200, both flat
|
|
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
|
|
|
|
|
|
def test_rebalance_skips_in_position_worker(tmp_path):
|
|
L = PortfolioLedger("PX", total_capital=1000.0, data_dir=tmp_path)
|
|
workers = {"a": FakeWorker(700.0, in_position=True), "b": FakeWorker(500.0)}
|
|
rebalance_allocations(L, workers, {"a": 0.5, "b": 0.5})
|
|
assert L.total_capital == 1200.0
|
|
assert workers["a"].capital == 700.0 # invariato: posizione aperta
|
|
assert workers["b"].capital == 600.0 # flat: riallineato
|