- Fix typo
- Optimize import order
- Aggregate the assertion tests
- Modify `IDFModel.transform` API for pyspark