[#86] Add test code