def test_get_consumption_by_product(self): values = LedgerBalancesIndicator._get_values_by_product( 'consumption', 'case1', ['coke', 'surge', 'new_coke'] ) self.assertEqual(values['coke'], 63) self.assertEqual(values['surge'], 0) self.assertEqual(values['new_coke'], 0)
def test_get_consumption_by_product(self): values = LedgerBalancesIndicator._get_values_by_product( 'consumption', self.case_id, ['coke', 'surge', 'new_coke'], self.domain_name) self.assertEqual(values['coke'], 63) self.assertEqual(values['surge'], 0) self.assertEqual(values['new_coke'], 0)
def test_get_soh_values_by_product(self): values = LedgerBalancesIndicator._get_values_by_product( 'soh', 'case1', ['coke', 'surge', 'new_coke'] ) self.assertEqual(values['coke'], 32) self.assertEqual(values['surge'], 85) self.assertEqual(values['new_coke'], 0)
def test_get_soh_values_by_product(self): values = LedgerBalancesIndicator._get_values_by_product( 'soh', self.case_id, ['coke', 'surge', 'new_coke'], self.domain_name) self.assertEqual(values['coke'], 32) self.assertEqual(values['surge'], 85) self.assertEqual(values['new_coke'], 0)
def _build_ledger_balances_indicator(spec, context): wrapped_spec = LedgerBalancesIndicatorSpec.wrap(spec) return LedgerBalancesIndicator(wrapped_spec)