EG: made test assert proper
EG: added some comments to understand it better.