Add direct heat and chiller tests