Write pytest tests for data.loader
I haven't copied the test part for data.loader to pytest because of the transformers requirement + slow tests. We might want some extra tests here, and move the transformers tests to a repo that has transformers as requirement (integration test for plugin_templates)