Add simple test for data import