Add tests for different sharding types (#1673) #5942

Job	Run time
build_test (linux.2xlarge, 3.10, py310) / linux-job	8m 11s
build_test (linux.2xlarge, 3.11, py311) / linux-job	8m 23s
build_test (linux.2xlarge, 3.9, py39) / linux-job	8m 29s
build_test (linux.2xlarge, 3.8, py38) / linux-job	7m 50s
	32m 53s

Provide feedback