self.assertAllClose(out[1][0][0], out[0][0][3])