Add tests for WeakType