Should add an unit test when loading an non-existent dataset, maybe using `unitest.assertraises`.