Add tests for partial import