Switch use of HasTraits.set to HasTraits.trait_set in tests

Remove uses of `HasTraits.set` in tests now that the method no longer exists.

Commit message is wrong. Shoudl read `trait_set`, not `trait_get`.