Removed center argument from shapes tests
Added tests for data generation