def test_variable_count(self): from libcellml import Component, Variable # size_t variableCount() c = Component() self.assertEqual(c.variableCount(), 0) c.addVariable(Variable()) self.assertEqual(c.variableCount(), 1) c.addVariable(Variable()) self.assertEqual(c.variableCount(), 2) c.removeVariable('') self.assertEqual(c.variableCount(), 1) c.removeVariable('') self.assertEqual(c.variableCount(), 0)
def test_remove_variable(self): from libcellml import Component, Variable # bool removeVariable(size_t index) c = Component() self.assertFalse(c.removeVariable(0)) self.assertFalse(c.removeVariable(-1)) self.assertFalse(c.removeVariable(1)) c.addVariable(Variable()) self.assertFalse(c.removeVariable(-1)) self.assertFalse(c.removeVariable(1)) self.assertTrue(c.removeVariable(0)) self.assertFalse(c.removeVariable(0)) del c # bool removeVariable(const std::string &name) c = Component() self.assertFalse(c.removeVariable('')) v1 = Variable() c.addVariable(v1) self.assertTrue(c.removeVariable('')) self.assertFalse(c.removeVariable('')) name = 'blue' v1.setName(name) self.assertFalse(c.removeVariable(name)) c.addVariable(v1) self.assertTrue(c.removeVariable(name)) self.assertFalse(c.removeVariable(name)) del [c, v1, name] # bool removeVariable(const VariablePtr &variable) c = Component() v1 = Variable("second") v2 = Variable("meter") self.assertFalse(c.removeVariable(v1)) c.addVariable(v1) self.assertFalse(c.removeVariable(v2)) self.assertTrue(c.removeVariable(v1)) self.assertFalse(c.removeVariable(v1))