def test_get_asset_count_and_cost_group_by_device(self):
venture = utils.get_or_create_venture()
device1 = utils.get_or_create_device()
device2 = utils.get_or_create_device()
utils.get_or_create_dailydevice(
datetime.date(2013, 10, 10),
device1,
venture,
price=1460,
deprecation_rate=25,
is_deprecated=False,
)
utils.get_or_create_dailydevice(
datetime.date(2013, 10, 11),
device2,
venture,
price=1460,
deprecation_rate=25,
is_deprecated=False,
)
utils.get_or_create_dailydevice(
datetime.date(2013, 10, 12),
device2,
self.venture1,
price=1460,
deprecation_rate=25,
is_deprecated=False,
)
result = Deprecation.get_assets_count_and_cost(
start=datetime.date(2013, 10, 10),
end=datetime.date(2013, 10, 20),
ventures=[venture],
group_by='pricing_device',
)
self.assertEquals(list(result), [
{
'assets_cost': D('1'),
'assets_count': 1.0,
'pricing_device': device1.id,
},
{
'assets_cost': D('1'),
'assets_count': 1.0,
'pricing_device': device2.id,
}
])
def test_get_asset_count_and_cost(self):
result = Deprecation.get_assets_count_and_cost(
start=datetime.date(2013, 10, 10),
end=datetime.date(2013, 10, 20),
ventures=self.ventures_subset,
)
self.assertEquals(list(result), [
{
'assets_cost': D('55'), # 1 * 5 (devices) * 11 (days)
'assets_count': 55.0, # 5 (devices) * 11 (days)
'pricing_venture': self.venture1.id,
},
{
'assets_cost': D('220'), # 2 * 10 (devices) * 11 (days)
'assets_count': 110.0, # 10 (devices) * 11 (days)
'pricing_venture': self.venture2.id,
}
])
