updating tests for dataset and metric refactor