# /// script# requires-python = ">=3.11"# dependencies = [#     "zarr>3",# ]# ///importzarrfromzarr.storageimportMemoryStorefromzarr.core.bufferimportdefault_buffer_prototypeimportasynciomem_a= {}mem_b= {}store_a=MemoryStore(mem_a)store_b=MemoryStore(mem_b)# create an array and fill it with valuesarray=zarr.create_array(store_a,name='foo',shape=(10,),dtype='float32')array[:]=1# define an async function that lists the contents of the source store and copies each key, value pair to the dest storeasyncdefcopy_store(store_a,store_b):asyncforxinstore_a.list():awaitstore_b.set(x,awaitstore_a.get(x,prototype=default_buffer_prototype()))print(f'setting{x} ')returnNone# run that async functionasyncio.run(copy_store(store_a,store_b))# check that the two dictionaries are identical, i.e. everything from mem_a is in mem_bassertmem_a==mem_b