Add missing type check before cast in TF model parsing test (small change)

[khatchad]

Small fix to check a cast in the test code. This is a small change, but a larger fix that we are preparing will depend on it.