Add tests for split data functionality