def test_read_text_by_regex(self):
text = '<b><font size="4"> 09/06/2017</b></font>'
date_text = IN_PB.read_text_by_regex('(\d+/\d+/\d+)', text)
expected = "09/06/2017"
self.assertEquals(date_text, expected)
text = '<b>Time : 14:38:29</b>'
date_text = IN_PB.read_text_by_regex('(\d+:\d+:\d+)', text)
expected = "14:38:29"
self.assertEquals(date_text, expected)
text = '<b> Last Updated at 13:33:59</b><br>'
date_text = IN_PB.read_text_by_regex('(\d+:\d+:\d+)', text)
expected = "13:33:59"
self.assertEquals(date_text, expected)
def test_read_text_by_regex(self):
text ='<b><font size="4"> 09/06/2017</b></font>'
date_text = IN_PB.read_text_by_regex('(\d+/\d+/\d+)', text)
expected = "09/06/2017"
self.assertEqual(date_text, expected)
text = '<b>Time : 14:38:29</b>'
date_text = IN_PB.read_text_by_regex('(\d+:\d+:\d+)', text)
expected = "14:38:29"
self.assertEqual(date_text, expected)
text = '<b> Last Updated at 13:33:59</b><br>'
date_text = IN_PB.read_text_by_regex('(\d+:\d+:\d+)', text)
expected = "13:33:59"
self.assertEqual(date_text, expected)
def test_read_punjab_consumption_date(self):
current = get(datetime(2018, 5, 5, 14, 38, 29), 'Asia/Kolkata')
date_text = "02/04/2018"
time_text = "14:38:29"
expected = get(datetime(2018, 4, 2, 14, 38, 29), 'Asia/Kolkata').datetime
date_time = IN_PB.read_punjab_consumption_date(date_text, time_text, current)
self.assertEqual(date_time, expected)
def test_time_string_to_kolkata_date(self):
utc_actual = get(datetime(2017, 9, 6, 12, 30, 0), 'UTC')
time_text = "12:25:30"
time_format = "HH:mm:ss"
date_time = IN_PB.time_string_to_kolkata_date(utc_actual, time_text,
time_format)
self.assertIsNotNone(date_time)
expected = get(datetime(2017, 9, 6, 12, 25, 30), 'Asia/Kolkata')
self.assertEquals(date_time, expected)
utc_actual = get(datetime(2017, 9, 7, 0, 5, 0), 'UTC')
time_text = "23:55:30"
time_format = "HH:mm:ss"
date_time = IN_PB.time_string_to_kolkata_date(utc_actual, time_text,
time_format)
self.assertIsNotNone(date_time)
expected = get(datetime(2017, 9, 6, 23, 55, 30), 'Asia/Kolkata')
self.assertEquals(date_time, expected)
def test_read_punjab_consumption_date(self):
current = get(datetime(2018, 5, 5, 14, 38, 29), 'Asia/Kolkata')
date_text = "02/04/2018"
time_text = "14:38:29"
expected = get(datetime(2018, 4, 2, 14, 38, 29), 'Asia/Kolkata').datetime
date_time = IN_PB.read_punjab_consumption_date(date_text, time_text, current)
self.assertEqual(date_time, expected)
def test_date_time_strings_to_kolkata_date(self):
date_text = "09/06/2017"
date_format = "DD/MM/YYYY"
time_text = "13:33:59"
time_format = "HH:mm:ss"
date_time = IN_PB.date_time_strings_to_kolkata_date(date_text, date_format, time_text, time_format)
self.assertIsNotNone(date_time)
expected = get(datetime(2017, 6, 9, 13, 33, 59), 'Asia/Kolkata')
self.assertEquals(date_time, expected)
def test_date_time_strings_to_kolkata_date(self):
date_text = "09/06/2017"
date_format = "DD/MM/YYYY"
time_text = "13:33:59"
time_format = "HH:mm:ss"
date_time = IN_PB.date_time_strings_to_kolkata_date(
date_text, date_format, time_text, time_format)
self.assertIsNotNone(date_time)
expected = get(datetime(2017, 6, 9, 13, 33, 59), 'Asia/Kolkata')
self.assertEquals(date_time, expected)
def test_fetch_production(self):
response_text = resource_string("parsers.test.mocks", "IN_PB_pbGenReal.html")
self.adapter.register_uri("GET", "http://www.punjabsldc.org/pungenrealw.asp?pg=pbGenReal",
content=response_text)
try:
data = IN_PB.fetch_production('IN-PB', self.session)
self.assertIsNotNone(data)
self.assertEqual(data['zoneKey'], 'IN-PB')
self.assertEqual(data['source'], 'punjasldc.org')
self.assertIsNotNone(data['datetime'])
self.assertIsNotNone(data['production'])
self.assertEqual(data['production']['hydro'], 554.0)
self.assertIsNotNone(data['storage'])
except Exception as ex:
self.fail("IN_KA.fetch_production() raised Exception: {0}".format(ex.message))
def test_fetch_consumption(self):
response_text = resource_string("parsers.test.mocks", "IN_PB_nrGenReal.html")
self.adapter.register_uri("GET", "http://www.punjabsldc.org/nrrealw.asp?pg=nrGenReal",
content=response_text)
try:
data = IN_PB.fetch_consumption('IN-PB', self.session)
self.assertIsNotNone(data)
self.assertEqual(data['zoneKey'], 'IN-PB')
self.assertEqual(data['source'], 'punjasldc.org')
self.assertIsNotNone(data['datetime'])
expected = get(datetime(2017, 9, 6, 14, 38, 29), 'Asia/Kolkata').datetime
date_time = data['datetime']
self.assertEqual(date_time, expected)
self.assertIsNotNone(data['consumption'])
self.assertEqual(data['consumption'], 7451.0)
except Exception as ex:
self.fail("IN_KA.fetch_consumption() raised Exception: {0}".format(ex.message))
def test_fetch_consumption(self):
response_text = resource_string("parsers.test.mocks", "IN_PB_nrGenReal.html")
self.adapter.register_uri("GET", "http://www.punjabsldc.org/nrrealw.asp?pg=nrGenReal",
content=response_text)
try:
data = IN_PB.fetch_consumption('IN-PB', self.session)
self.assertIsNotNone(data)
self.assertEqual(data['zoneKey'], 'IN-PB')
self.assertEqual(data['source'], 'punjasldc.org')
self.assertIsNotNone(data['datetime'])
expected = get(datetime(2017, 9, 6, 14, 38, 29), 'Asia/Kolkata').datetime
date_time = data['datetime']
self.assertEqual(date_time, expected)
self.assertIsNotNone(data['consumption'])
self.assertEqual(data['consumption'], 7451.0)
except Exception as ex:
self.fail("IN_KA.fetch_consumption() raised Exception: {0}".format(ex.message))
def test_fetch_production(self):
response_text = resource_string("parsers.test.mocks", "IN_PB_pbGenReal.html")
solar_response_text = resource_string("parsers.test.mocks", "IN_PB_solarppW.html")
self.adapter.register_uri("GET", "http://www.punjabsldc.org/pungenrealw.asp?pg=pbGenReal",
content=response_text)
self.adapter.register_uri("GET", "http://www.punjabsldc.org/solarppW.asp?pg=solarppW",
content=solar_response_text)
try:
data = IN_PB.fetch_production('IN-PB', self.session)
self.assertIsNotNone(data)
self.assertEqual(data['zoneKey'], 'IN-PB')
self.assertEqual(data['source'], 'punjasldc.org')
self.assertIsNotNone(data['datetime'])
self.assertIsNotNone(data['production'])
self.assertEqual(data['production']['hydro'], 554.0)
self.assertIsNotNone(data['storage'])
except Exception as ex:
self.fail("IN_KA.fetch_production() raised Exception: {0}".format(ex.message))