def test_conditional_probability_expression(self):
from mitesting.customized_commands import conditional_probability_expression
from mitesting.sympy_customized import SymbolCallable
A=Symbol("A")
B=Symbol("B")
P=SymbolCallable("P")
expr=parse_expr("A|B")
self.assertEqual(expr, conditional_probability_expression(A,B))
self.assertEqual(latex(expr), "A ~|~ B")
expr=parse_expr("P(A|B)", local_dict={'P': P})
self.assertEqual(expr, P(conditional_probability_expression(A,B)))
self.assertEqual(latex(expr), "P{\\left (A ~|~ B \\right )}")
def test_absolute_values(self):
from sympy import Abs
x=Symbol('x')
y=Symbol('y')
z=Symbol('z')
self.assertEqual(parse_expr("|x+y|"), Abs(x+y))
self.assertEqual(parse_expr("x|y|x|z|x"), x**3*Abs(y)*Abs(z))
self.assertEqual(parse_expr("|x+y|"), Abs(x+y))
self.assertRaises(SyntaxError, parse_expr,"|x+|y||")
self.assertEqual(parse_expr("|x+(|y|)|"), Abs(x+Abs(y)))
from mitesting.customized_commands import conditional_probability_expression
from mitesting.sympy_customized import SymbolCallable
A=Symbol("A")
B=Symbol("B")
P=SymbolCallable("P")
expr=parse_expr("|P(A|B)|", local_dict={'P': P})
self.assertEqual(expr, Abs(P(conditional_probability_expression(A,B))))
