manyi/testcase/unit/test_stat.py

# -*- coding: utf-8 -*-
#!/usr/bin/env python

from bson.objectid import ObjectId
from apilib.monetary import Ratio, RMB

from apps.web.report.utils import record_income_stats, record_consumption_stats
from apps.web.dealer.proxy import DealerIncomeProxy
from apps.web.user.models import ConsumeRecord
from apps.web.report.ledger import Ledger

from apps.web.report.structures import GroupIncomePartition, SubIncomePartition


# base cases

def test_create_SubIncomePartition():

    assert SubIncomePartition(object_id=ObjectId(), role='owner', share=Ratio(100), money=RMB(10))

def test_create_GroupIncomePartition():

    owner_partition = [SubIncomePartition(object_id=ObjectId(), role='owner', share=Ratio(100), money=RMB(10))]
    partner_partition = []
    agent_partition = []

    assert GroupIncomePartition(money=RMB(10), owner=owner_partition, partner=partner_partition, agent=agent_partition)