def test_fill_in_missing_units_list(self):
"""Test that missing units are filled in."""
expected_unit = 'Hrs'
data = [{
'date': '2018-07-22',
'units': '',
'instance_type': 't2.micro',
'total': 30.0,
'count': 0
}, {
'date': '2018-07-22',
'units': expected_unit,
'instance_type': 't2.small',
'total': 17.0,
'count': 0
}, {
'date': '2018-07-22',
'units': expected_unit,
'instance_type': 't2.micro',
'total': 1.0,
'count': 0
}]
unit = _find_unit()(data)
result = _fill_in_missing_units(unit)(data)
for entry in result:
self.assertEqual(entry.get('units'), expected_unit)
def test_find_unit_list(self):
"""Test that the correct unit is returned."""
expected_unit = 'Hrs'
data = [{
'date': '2018-07-22',
'units': '',
'instance_type': 't2.micro',
'total': 30.0,
'count': 0
}, {
'date': '2018-07-22',
'units': expected_unit,
'instance_type': 't2.small',
'total': 17.0,
'count': 0
}, {
'date': '2018-07-22',
'units': expected_unit,
'instance_type': 't2.micro',
'total': 1.0,
'count': 0
}]
result_unit = _find_unit()(data)
self.assertEqual(expected_unit, result_unit)
def test_find_unit_dict(self):
"""Test that the correct unit is returned for a dictionary."""
data = {'date': '2018-07-22', 'units': '',
'instance_type': 'Standard_A1_v2', 'total': 30.0, 'count': 0},
result_unit = _find_unit()(data)
self.assertIsNone(result_unit)
def test_fill_in_missing_units_list(self):
"""Test that missing units are filled in."""
expected_unit = "Hrs"
data = [
# AWS
{"date": "2018-07-22", "units": "", "instance_type": "t2.micro", "total": 30.0, "count": 0},
{"date": "2018-07-22", "units": expected_unit, "instance_type": "t2.small", "total": 17.0, "count": 0},
{"date": "2018-07-22", "units": expected_unit, "instance_type": "t2.micro", "total": 1.0, "count": 0},
# Azure
{"date": "2018-07-22", "units": "", "instance_type": "Standard_A1_v2", "total": 30.0, "count": 0},
{
"date": "2018-07-22",
"units": expected_unit,
"instance_type": "Standard_A2m_v2",
"total": 17.0,
"count": 0,
}, # noqa: E501
{
"date": "2018-07-22",
"units": expected_unit,
"instance_type": "Standard_A1_v2",
"total": 1.0,
"count": 0,
},
]
unit = _find_unit()(data)
result = _fill_in_missing_units(unit)(data)
for entry in result:
self.assertEqual(entry.get("units"), expected_unit)
def test_find_unit_list(self):
"""Test that the correct unit is returned."""
expected_unit = "Hrs"
data = [
{
"date": "2018-07-22",
"units": "",
"instance_type": "t2.micro",
"total": 30.0,
"count": 0
},
{
"date": "2018-07-22",
"units": expected_unit,
"instance_type": "t2.small",
"total": 17.0,
"count": 0
},
{
"date": "2018-07-22",
"units": expected_unit,
"instance_type": "t2.micro",
"total": 1.0,
"count": 0
},
]
result_unit = _find_unit()(data)
self.assertEqual(expected_unit, result_unit)
def test_find_unit_dict(self):
"""Test that the correct unit is returned for a dictionary."""
data = {
"date": "2018-07-22",
"units": "",
"instance_type": "t2.micro",
"total": 30.0,
"count": 0
}
result_unit = _find_unit()(data)
self.assertIsNone(result_unit)
