add usage of from_dataset in tests where possible