コード例 #1
0
ファイル: tests.py プロジェクト: danst0/Portfolio
    def test_list_pf(self):
        pf = Portfolio.objects.create(name='Test')
        sec = Security()
        price = Price()
        mysec1 = sec.add('Test', ['SomeAliasString'], '', 'APC.DE', 'Stock')
        mysec2 = sec.add('Test2', ['SomeAliasString'], '', 'APC.DE', 'Stock')
        time_price_1 = timezone.now() - datetime.timedelta(days=20)
        time_price_2 = timezone.now() - datetime.timedelta(days=5)
        time_price_3 = timezone.now() - datetime.timedelta(days=2)
        time_price_4 = timezone.now() - datetime.timedelta(days=1)
        price.add(mysec1, time_price_1, 100)
        price.add(mysec1, time_price_2, 15)
        price.add(mysec1, time_price_3, 15)
        price.add(mysec1, time_price_4, 10)

        price.add(mysec2, time_price_1, 40)
        price.add(mysec2, time_price_2, 50)
        price.add(mysec2, time_price_3, 60)
        price.add(mysec2, time_price_4, 70)
        t = Transaction()
        # def add(self, transaction_type, portfolio, stock_id, date, nominal, price, cost):
        t.add('b', pf, mysec1, time_price_1, 10, 100, 0)
        t.add('b', pf, mysec1, time_price_2, 10, 50, 0)
        t.add('s', pf, mysec1, time_price_3, 10, 15, 0)
        t.add('b', pf, mysec2, time_price_1, 10, 40, 0)
        result = t.list_pf(pf.name, time_price_1, time_price_4)
        # print()
        # for item in result:
        #     print(item['name'], item['profit'])
        # print()
        self.assertEqual(result[0]['profit'], Decimal('-1250'))
        self.assertEqual(result[1]['profit'], Decimal('300'))
        self.assertEqual(result[2]['profit'], Decimal('-950'))
コード例 #2
0
ファイル: tests.py プロジェクト: danst0/Portfolio
    def a_test_portfolio_overview(self):
        sec = Security.objects.create(name='TestSec', aliases='Test Alias', isin_id='DETest', yahoo_id='ABC',
                                      type='Stock')
        pf = Portfolio.objects.create(name='Test')
        t = Transaction()
        time = timezone.now().date()
        p = Price()
        p.add(sec, time, Decimal(111))
        t.add('b', pf, sec, time, 100, 100, 100)

        p = t.list_pf(pf.name, time)
        self.assertEqual(p, [{
                                 'nominal': Decimal('100'),
                                 'invest': Decimal('-10100'),
                                 'profit': Decimal('1000'),
                                 'name': 'TestSec',
                                 'value': Decimal('11100'),
                                 'price': Decimal('111'),
                                 'cost': Decimal('100')},
                             {
                                 'name': 'Total',
                                 'value': Decimal('11100')}])