Revisit tests

[zpervan]

A lot of tests are currently testing plot data position which is prone to fail if something in the configuration fails. Write tests in a way so that they test correct number of shapes based on plot data.