def test_setitem_with_cuba_key(self):
     container = DataContainer()
     container[CUBA("POTENTIAL_ENERGY")] = 29
     self.assertIsInstance(container.keys()[0], CUBA)
     self.assertEqual(container[CUBA("POTENTIAL_ENERGY")], 29)
 def test_setitem_with_cuba_key(self):
     container = DataContainer()
     container[CUBA("POTENTIAL_ENERGY")] = 29
     self.assertIsInstance(container.keys()[0], CUBA)
     self.assertEqual(container[CUBA("POTENTIAL_ENERGY")], 29)