Added tests to tn models