def test_get_consumption_by_product(self): values = get_values_by_product( self.domain_name, self.case_id, 'consumption', ['coke', 'surge', 'new_coke'] ) self.assertEqual(values.get('coke'), 63) self.assertEqual(values.get('surge'), None) self.assertEqual(values.get('new_coke'), None)
def test_get_soh_values_by_product(self): values = get_values_by_product( self.domain_name, self.case_id, 'soh', ['coke', 'surge', 'new_coke'] ) self.assertEqual(values.get('coke'), 32) self.assertEqual(values.get('surge'), 85) self.assertEqual(values.get('new_coke'), None)
def test_get_consumption_by_product(self): values = get_values_by_product(self.domain_name, self.case_id, 'consumption', ['coke', 'surge', 'new_coke']) self.assertEqual(values.get('coke'), 63) self.assertEqual(values.get('surge'), None) self.assertEqual(values.get('new_coke'), None)
def _get_values_by_product(self, domain, case_id): return get_values_by_product(domain, case_id, self.ledger_section, self.product_codes)
def _get_values_by_product(self, domain, case_id): return get_values_by_product(domain, case_id, self.ledger_section, self.product_codes)
def test_get_soh_values_by_product(self): values = get_values_by_product(self.domain_name, self.case_id, 'soh', ['coke', 'surge', 'new_coke']) self.assertEqual(values.get('coke'), 32) self.assertEqual(values.get('surge'), 85) self.assertEqual(values.get('new_coke'), None)