aboutsummaryrefslogtreecommitdiffstats
path: root/python/tests/test_cds.py
blob: a01fbe16cfa01b71546f1b41f0d860f7998b294b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
import unittest
from pyisda.cdsone import upfront_charge
from pyisda.curve import SpreadCurve
from pyisda.legs import ContingentLeg, FeeLeg
from pyisda.date import roll_date
import datetime
import numpy as np
from quantlib.settings import Settings
from quantlib.time.api import Date

import sys
sys.path.append('..')
from analytics import CreditIndex
from yieldcurve import YC, ql_to_jp, get_curve

class TestUpfront(unittest.TestCase):
    index = CreditIndex("ig", 26, "5yr",
                        value_date=datetime.date(2016, 9, 21))
    index.notional = 50e6
    index.spread = 70

    def test_upfront(self):
        self.assertAlmostEqual(-self.index.pv, 685292.81, 2)

    def test_cdsone(self):
        jp_yc = get_curve(self.index.value_date)
        fee_dirty = self.index.notional * (
            upfront_charge(self.index.value_date, self.index._cash_settle_date,
                           self.index.start_date,
                           self.index._step_in_date, self.index.start_date,
                           self.index.end_date, self.index.fixed_rate*1e-4,
                           jp_yc, 70e-4, self.index.recovery,
                           False))
        fee_clean = self.index.notional * (
            upfront_charge(self.index.value_date, self.index._cash_settle_date,
                           self.index.start_date,
                           self.index._step_in_date, self.index.start_date,
                           self.index.end_date, self.index.fixed_rate*1e-4,
                           jp_yc, 70e-4, self.index.recovery,
                           True))
        self.assertAlmostEqual(-fee_dirty, 685292.81, 2)
        self.assertAlmostEqual(fee_clean, self.index.clean_pv)

    def test_annuity(self):
        self.assertAlmostEqual(-self.index.clean_pv,
                               self.index.notional * self.index.risky_annuity *
                               (self.index.fixed_rate - self.index.spread)*1e-4)

class TestSpreadCurve(unittest.TestCase):

    def setUp(self):
        self.upfront_curve = 1e-2 * np.array(
            [-2.502394, -4.871879, -9.329793, -12.98734, -15.833254,
             -17.622571, -20.505054, -24.314297])
        spread_curve = 1e-4 * np.array([500, 500, 500, 500, 500, 500, 500, 500])
        recovery_curve = np.full(8, 0.3)

        self.trade_date = datetime.date(2018, 6, 18)
        self.step_in_date = datetime.date(2018, 6, 19)
        self.cash_settle_date = datetime.date(2018, 6, 21)

        self.yc = get_curve(self.trade_date, "USD")
        self.tenors = np.array((0.5, 1, 2, 3, 4, 5, 7, 10))
        self.sc = SpreadCurve(self.trade_date, self.yc, None, None, None,
                              self.tenors, spread_curve, self.upfront_curve, recovery_curve, ticker="AES")

    def test_upfront_curves(self):

        maturities = [roll_date(self.trade_date, t) for t in self.tenors]
        for m, upf in zip(maturities, self.upfront_curve):
            pl = ContingentLeg(datetime.date(2017, 9, 20), m, 1)
            cl = FeeLeg(datetime.date(2017, 9, 20), m, True, 1, 0.05)
            a = pl.pv(self.trade_date, self.step_in_date, self.cash_settle_date, self.yc, self.sc, 0.3)
            b = cl.pv(self.trade_date, self.step_in_date, self.cash_settle_date, self.yc, self.sc, True)
            self.assertAlmostEqual(a - b, upf)

    def test_roundtrip(self):
        sc_copy = SpreadCurve.from_bytes(self.sc.as_buffer(True), True)
        self.assertEqual(sc_copy.inspect(), self.sc.inspect())
        self.assertEqual(sc_copy.full_ticker, self.sc.full_ticker)
        sc_copy = SpreadCurve.from_bytes(self.sc.as_buffer(False), False)
        self.assertEqual(sc_copy.inspect(), self.sc.inspect())
        self.assertEqual(sc_copy.full_ticker, self.sc.full_ticker)

if __name__ == "__main__":
    unittest.main()