Add tests for DataImport