Added test for input and output dim
Added test for computeZ
Added tests for Predict