def test_formula_get(self): f = Formula({Atom.H: 12, Atom.C: 6, Atom.O: 6}) self.assertEqual(f.get(Atom.H), 12) self.assertEqual(f.get(Atom.Au), None) self.assertEqual(f.get(Atom.Hg, 4), 4)