CDK-437. Input format doesn't work with datasets with more than one partition level.
Make test for keyFromDirectory more stable.
Use original dataset (not temporary) when getting partition key.