energy-calc/tests.py at master · CenterForTheBuiltEnvironment/energy-calc · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
import main
import unittest
import json

class EnergyCalcTestCase(unittest.TestCase):

    def setUp(self):
        self.app = energycalc.app.test_client()

    def tearDown(self):
        pass

    def api(self, hsp0, hsp1, csp0, csp1, climate):
        url = '/api?climate=%s&hsp0=%s&hsp1=%s&csp0=%s&csp1=%s' \
                % (climate, hsp0, hsp1, csp0, csp1)
        rv = self.app.get(url)
        return rv

    def test_root(self):
        rv = self.app.get('/')
        assert rv.status_code == 200

    def test_api(self):
        rv = self.app.get('/api?climate=San%20Francisco&hsp0=70&hsp1=68&csp0=72&csp1=76.3')
        assert rv.status_code == 200

    def test_basecase(self):
        rv = self.api(70, 70, 72, 72, 'San Francisco')
        assert rv.status_code == 200
        data = json.loads(rv.data)
        nonzero_cooling = filter(lambda v: v != 0.0, data['cooling']['chart_data'].values())
        assert len(nonzero_cooling) == 0, "Cooling savings are not zero"
        nonzero_heating = filter(lambda v: v != 0.0, data['heating']['chart_data'].values())
        assert len(nonzero_heating) == 0, "Cooling savings are not zero"

    def test_hsp_truncate(self):
        hsp0, hsp1, csp0, csp1 = 73, 71, 74, 74
        setpoints = (hsp0, hsp1, csp0, csp1)
        rv = self.api(hsp0, hsp1, csp0, csp1, 'San Francisco')
        assert rv.status_code == 200
        data = json.loads(rv.data)
        nonzero_cooling = filter(lambda v: v != 0.0, data['cooling']['chart_data'].values())
        assert len(nonzero_cooling) == 0, \
                "Cooling savings are not zero: hsp0=%s, hsp1=%s, csp0=%s, csp1=%s" % setpoints
        nonzero_heating = filter(lambda v: v != 0.0, data['heating']['chart_data'].values())
        assert len(nonzero_heating) == 0, \
                "Heating savings are not zero: hsp0=%s, hsp1=%s, csp0=%s, csp1=%s" % setpoints

if __name__ == '__main__':
    unittest.main()