[minor] Adding tests for the spark interface. Make it more obvious where the data is flowing to.