Updated tests, now imported to plotnik.py as module