checking data.shape for each test dataset