Add tests for split data functionality
Add tests for split data functionality