import pylab in unit test area