Update test_tnt.py