def testGivenExpressionWithTwoVariablesWithDifferentNameWhenGetNameSetThenSetWithVariablesNamesIsReturned(
self):
variable1 = VariableBuilder().name('x').build()
variable2 = VariableBuilder().name('y').build()
expression = ExpressionBuilder().term(variable1).term(
variable2).build()
self.assertEqual(expression.get_name_set(),
{variable1.name, variable2.name})
def testGivenExpressionWithOneVariableWhenAddExpressionWithOtherVariableAndConstantThenExpressionContainsAllTerms(
self):
variable1 = VariableBuilder().name('x').build()
variable2 = VariableBuilder().name('y').build()
constant = ConstantBuilder().value(1.0).build()
expression1 = ExpressionBuilder().term(variable1).build()
expression2 = ExpressionBuilder().term(variable2).term(
constant).build()
expression1.add_expression(expression2)
self.assertEqual(expression1.get_name_set(),
{variable1.name, variable2.name})
self.assertEqual(expression1.get_value_constant(), constant.value)
def testGivenExpressionWithConstantOnlyWhenGetNameSetThenEmptySetIsReturned(
self):
expression = ExpressionBuilder().default_constant().build()
self.assertEqual(expression.get_name_set(), set())
def testGivenExpressionWithTwoVariablesWithSameNameWhenGetNameSetThenSetWithVariablesNameIsReturned(
self):
variable = VariableBuilder().name('x').build()
expression = ExpressionBuilder().term(variable).term(
variable.clon()).build()
self.assertEqual(expression.get_name_set(), {variable.name})
def testGivenEmptyExpressionWhenGetNameSetThenEmptySetIsReturned(self):
expression = ExpressionBuilder().build()
self.assertEqual(expression.get_name_set(), set())
