sample_python_sympy.py 306 B

123456789101112131415
  1. #! /usr/bin/env python3
  2. from sympy import symbols, expand, factor
  3. x, y = symbols('x y')
  4. expr = x + 2*y
  5. expanded_expr = expand(x*expr)
  6. print(expanded_expr)
  7. assert str(expanded_expr) == "x**2 + 2*x*y"
  8. factored_expr = factor(expanded_expr)
  9. print(factored_expr)
  10. assert str(factored_expr) == "x*(x + 2*y)"