api_rte/tests/test_api.py at 9cd63db141bd1069396ec95bc44424308cd4a8f2 · ReinboldV/api_rte · 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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
import unittest
from api_rte.config import *
from api_rte.api import *
import datetime


class TestApiRte(unittest.TestCase):

    def test_get_token(self):
        token_type, token = get_token(client_id=CLIENT_ID, client_secret=CLIENT_SECRET, oauth_url=OAUTH_URL)
        self.assertIsNotNone(token, msg='Authentication failed')
        self.assertIsNotNone(token_type, msg='Authentication failed')

    def test_get_prod_json(self):
        token_type, token = get_token(client_id=CLIENT_ID, client_secret=CLIENT_SECRET, oauth_url=OAUTH_URL)

        TSTART = datetime.datetime.today() - datetime.timedelta(days=2)
        TEND = datetime.datetime.today()

        r = get_production_json(start_date=TSTART, end_date=TEND, token_type=token_type, access_token=token)
        self.assertTrue('forecasts' in r.keys())
        self.assertFalse('error' in r.keys())

        r = get_production_json(start_date=TSTART,
                                end_date=TEND,
                                token_type=token_type,
                                access_token=token,
                                production_type='WIND',
                                type='D-1')
        self.assertTrue('forecasts' in r.keys())
        self.assertFalse('error' in r.keys())

        r = get_production_json(start_date=TSTART,
                                end_date=TEND,
                                token_type=token_type,
                                access_token=token,
                                production_type='WIND',
                                type='ID')
        self.assertTrue('forecasts' in r.keys())
        self.assertFalse('error' in r.keys())

        r = get_production_json(start_date=TSTART,
                                end_date=TEND,
                                token_type=token_type,
                                access_token=token,
                                production_type='WIND',
                                type='CURRENT')
        self.assertTrue('forecasts' in r.keys())
        self.assertFalse('error' in r.keys())

        r = get_production_json(start_date=TSTART,
                                end_date=TEND,
                                token_type=token_type,
                                access_token=token,
                                production_type='SOLAR',
                                type='D-1')
        self.assertTrue('forecasts' in r.keys())
        self.assertFalse('error' in r.keys())

        r = get_production_json(start_date=TSTART,
                                end_date=TEND,
                                token_type=token_type,
                                access_token=token,
                                production_type='SOLAR',
                                type='CURRENT')
        self.assertTrue('forecasts' in r.keys())
        self.assertFalse('error' in r.keys())

        r = get_production_json(start_date=TSTART,
                                end_date=TEND,
                                token_type=token_type,
                                access_token=token,
                                production_type='SOLAR',
                                type='ID')
        self.assertTrue('forecasts' in r.keys())
        self.assertFalse('error' in r.keys())

    def test_get_tempo_json(self):
        token_type, token = get_token(client_id=CLIENT_ID, client_secret=CLIENT_SECRET, oauth_url=OAUTH_URL)

        TSTART = datetime.datetime.today() - datetime.timedelta(days=2)
        TEND = datetime.datetime.today() + datetime.timedelta(days=1)
        r = get_tempo_json(start_date=TSTART, end_date=TEND, token_type=token_type, access_token=token)
        self.assertTrue('tempo_like_calendars' in r.keys())
        self.assertFalse('error' in r.keys())

    def test_parse_production(self):
        token_type, token = get_token(client_id=CLIENT_ID, client_secret=CLIENT_SECRET, oauth_url=OAUTH_URL)

        TSTART = datetime.datetime.today() - datetime.timedelta(days=2)
        TEND = datetime.datetime.today() + datetime.timedelta(days=1)
        r = get_production_json(start_date=TSTART, end_date=TEND, token_type=token_type, access_token=token)
        d = parse_production(r)
        self.assertTrue(isinstance(d, list))
        self.assertTrue(isinstance(d[0], dict))

    def test_get_prod(self):
        TSTART = datetime.datetime.today() - datetime.timedelta(days=2)
        TEND = datetime.datetime.today() + datetime.timedelta(days=0)

        r = get_prod(TSTART, TEND, production_type='SOLAR', type='D-1')
        self.assertTrue(isinstance(r, list))
        self.assertTrue(isinstance(r[0], dict))
        self.assertTrue(isinstance(r[0]['values'], pd.DataFrame))

        r = get_prod(TSTART, TEND, production_type='SOLAR', type='ID')
        self.assertTrue(isinstance(r, list))
        self.assertTrue(isinstance(r[0], dict))
        self.assertTrue(isinstance(r[0]['values'], pd.DataFrame))

        r = get_prod(TSTART, TEND, production_type='SOLAR', type='CURRENT')
        self.assertTrue(isinstance(r, list))
        self.assertTrue(isinstance(r[0], dict))
        self.assertTrue(isinstance(r[0]['values'], pd.DataFrame))


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