Skip to content

Commit aa47e86

Browse files
committed
update run pool UT
1 parent 1db8335 commit aa47e86

File tree

1 file changed

+17
-9
lines changed

1 file changed

+17
-9
lines changed

absbox/tests/regression/test_main.py

Lines changed: 17 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -227,16 +227,24 @@ def test_asset_02(setup_api):
227227
assert ppyRate(r[0])[1:] == [0.02, 0.02, 0.02]+[0.03]*6
228228

229229

230-
#@pytest.mark.pool
231-
#def test_pool_01(setup_api):
232-
# r = setup_api.runPool(myPool
233-
# ,poolAssump=("Pool",("Mortgage",None,None,None,None)
234-
# ,None
235-
# ,None)
236-
# ,read = True
237-
# ,breakdown = True )
230+
@pytest.mark.pool
231+
def test_pool_01(setup_api):
232+
r = setup_api.runPool(myPool
233+
,poolAssump=("Pool",("Mortgage",None,None,None,None)
234+
,None
235+
,None)
236+
,read = True
237+
,breakdown = False)
238+
assert 'breakdown' not in r['PoolConsol'], "breakdown should not be in PoolConsol"
239+
r = setup_api.runPool(myPool
240+
,poolAssump=("Pool",("Mortgage",None,None,None,None)
241+
,None
242+
,None)
243+
,read = True
244+
,breakdown = True)
245+
assert 'breakdown' in r['PoolConsol'], "breakdown should be in PoolConsol"
246+
assert 2 == len(r['PoolConsol']['breakdown']), "breakdown should have 2 items"
238247

239-
240248
@pytest.mark.analytics
241249
def test_first_loss(setup_api):
242250
r0 = setup_api.runFirstLoss(test01

0 commit comments

Comments
 (0)