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))