Update unit test to new temperature function