fixed tests for thermal storage model