add basic tests for months_after logic and fix it
Merge pull request #4088 from dimagi/opm-cz

add basic tests for months_after logic and fix it