Change the Test source file.

Signed-off-by: Pengfei Zhang <pzhang11@stevens.edu>