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.upfront, 0.)