def add_default_quantities(self):
"""
Adds any default symbols which are not present in the graph
Returns:
None
"""
new_syms = set(DEFAULT_SYMBOL_VALUES.keys())
new_syms -= set(self._symbol_to_quantity.keys())
for sym in new_syms:
quantity = Quantity.from_default(sym)
warnings.warn("Adding default {} quantity with value {}".format(
sym, quantity))
self.add_quantity(quantity)
def test_from_default(self):
default = Quantity.from_default('temperature')
self.assertEqual(default, Quantity('temperature', 300))
default = Quantity.from_default('relative_permeability')
self.assertEqual(default, Quantity("relative_permeability", 1))