Add ensure_test_case method.
Remove unused save.
Remove non-necessary check.