Python functional test for linux command

I want to write functional tests for linux sort command in python.I am new with writing test cases. While researching i found Cram and pytest that can be used but not sure. So any leads or links on how to proceed for writing functional test scripts for linux sort command would be appreciated