Add more tests for multi-word model associations

We can seperate tests for single and multi-word models