Subset is just a thin wrapper using the passed indices to index the underlying Dataset without actually manipulating it.
Have a look at this line of code 65.
You’ll get the right samples, if you index the Subset directly:
x, y = test_ds[0]