Tests shapes

Added tests for all the shapes.