Update test.py

import lxml