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