Add test for doctest case