[DCE_TEST] Add Input argument to take in number of frames to write.