Fix unit tests

[dmitriyrepin]

Fix unit tests: I forgot to update them with the

      # For now, we are hardcoding the chunk size to 1024.
      self._var_chunk_shape = (1, 1, 16, 1, 32, 1024)

change