Add Type model tests