2 files changed, 24 insertions, 0 deletions
diff --git a/python/tests/test_index.py b/python/tests/test_index.py
index 3b15cce8..6f127fcd 100644
--- a/python/tests/test_index.py
+++ b/python/tests/test_index.py
@@ -5,6 +5,7 @@ import numpy as np
 from pyisda.cdsone import upfront_charge
 
 from analytics import CreditIndex, ForwardIndex
+from analytics.basket_index import BasketIndex
 from analytics.index import g
 import pickle
 
@@ -18,6 +19,14 @@ class TestPickle(unittest.TestCase):
         a = pickle.loads(pickle.dumps(self.index))
         self.assertTrue(hash(a) == hash(self.index))
 
+    def test_pickle_basket(self):
+        a = BasketIndex("IG", 31, ["5yr"], value_date=datetime.date(2019, 2, 19))
+        self.assertTrue(hash(pickle.loads(pickle.dumps(a))), hash(a))
+
+    def test_from_tradeid(self):
+        ig28 = CreditIndex.from_tradeid(874)
+        self.assertTrue(ig28.spread, 68.)
+
 
 class TestStrike(unittest.TestCase):
     index = CreditIndex("ig", 26, "5yr", value_date=datetime.date(2016, 7, 1))
diff --git a/python/tests/test_tranche_basket.py b/python/tests/test_tranche_basket.py
new file mode 100644
index 00000000..e423db6c
--- /dev/null
+++ b/python/tests/test_tranche_basket.py
@@ -0,0 +1,15 @@
+import unittest
+import datetime
+from analytics import DualCorrTranche
+import pickle
+
+
+class TestPickle(unittest.TestCase):
+    bozeman = DualCorrTranche.from_tradeid(1037)
+
+    def test_pickle(self):
+        a = pickle.loads(pickle.dumps(self.bozeman))
+        self.assertTrue(hash(a) == hash(self.bozeman))
+
+    def test_pv(self):
+        self.assertAlmostEqual(self.bozeman.pv, 0.)