Add a test case for pooling and rnorm layer