[PATCH 2/2] [DONOTMERGE] tests: Add tests for SampleGrad() method