btrfs-progs: test/common: Enhance prepare_test_dev to reset device size