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