def test_shape(): X,y = genData(**args) # Equal length assert_equal(len(X),len(y)) assert_equal(len(X),n) assert_equal(X.shape[1],d) #assert_raises(ValueError,genData.genData,)