Update test_dataset_assembly.py
Add the mixture_bench parameter and the related function to better sample data classified into different types
Add the mixture_bench parameter and the related function to better sample data classified into different types