Add test cases for DS function node