Making DataPruner testable