def test_PI_holidays(self): pi_holidays = holidays.BR(state="PI") self.assertIn("2018-03-13", pi_holidays) self.assertEqual(pi_holidays[date(2018, 3, 13)], "Dia da Batalha do Jenipapo") self.assertIn("2018-10-19", pi_holidays) self.assertEqual(pi_holidays[date(2018, 10, 19)], "Dia do Piauí")
def test_SC_holidays(self): sc_holidays = holidays.BR(state="SC") self.assertIn("2018-08-11", sc_holidays) self.assertEqual( sc_holidays[date(2018, 8, 11)], "Criação da capitania, separando-se de SP", )
def test_SP_holidays(self): sp_holidays = holidays.BR(state="SP") self.assertIn("2018-07-09", sp_holidays) self.assertEqual( sp_holidays[date(2018, 7, 9)], "Revolução Constitucionalista de 1932", )
def test_DF_holidays(self): df_holidays = holidays.BR(state="DF") self.assertIn("2018-04-21", df_holidays) self.assertEqual(df_holidays[date(2018, 4, 21)], "Fundação de Brasília, Tiradentes") self.assertIn("2018-11-30", df_holidays) self.assertEqual(df_holidays[date(2018, 11, 30)], "Dia do Evangélico")
def test_PA_holidays(self): pa_holidays = holidays.BR(state="PA") self.assertIn("2018-08-15", pa_holidays) self.assertEqual( pa_holidays[date(2018, 8, 15)], "Adesão do Grão-Pará à independência do Brasil", )
def test_RN_holidays(self): rn_holidays = holidays.BR(state="RN") self.assertIn("2018-06-29", rn_holidays) self.assertEqual(rn_holidays[date(2018, 6, 29)], "Dia de São Pedro") self.assertIn("2018-10-03", rn_holidays) self.assertEqual(rn_holidays[date(2018, 10, 3)], "Mártires de Cunhaú e Uruaçuu")
def test_RJ_holidays(self): rj_holidays = holidays.BR(state="RJ") self.assertIn("2018-04-23", rj_holidays) self.assertEqual(rj_holidays[date(2018, 4, 23)], "Dia de São Jorge") self.assertIn("2018-10-28", rj_holidays) self.assertEqual(rj_holidays[date(2018, 10, 28)], "Dia do Funcionário Público") self.assertIn("2018-11-20", rj_holidays) self.assertEqual(rj_holidays[date(2018, 11, 20)], "Zumbi dos Palmares")
def test_PE_holidays(self): pe_holidays = holidays.BR(state="PE") self.assertIn("2018-03-06", pe_holidays) self.assertEqual( pe_holidays[date(2018, 3, 6)], "Revolução Pernambucana (Data Magna)", ) self.assertIn("2018-06-24", pe_holidays) self.assertEqual(pe_holidays[date(2018, 6, 24)], "São João")
def test_MA_holidays(self): ma_holidays = holidays.BR(state="MA") self.assertIn("2018-07-28", ma_holidays) self.assertEqual( ma_holidays[date(2018, 7, 28)], "Adesão do Maranhão à independência do Brasil", ) self.assertIn("2018-12-08", ma_holidays) self.assertEqual(ma_holidays[date(2018, 12, 8)], "Dia de Nossa Senhora da Conceição")
def test_AP_holidays(self): ap_holidays = holidays.BR(state="AP") self.assertIn("2018-03-19", ap_holidays) self.assertEqual(ap_holidays[date(2018, 3, 19)], "Dia de São José") self.assertIn("2018-07-25", ap_holidays) self.assertEqual(ap_holidays[date(2018, 7, 25)], "São Tiago") self.assertIn("2018-10-05", ap_holidays) self.assertEqual(ap_holidays[date(2018, 10, 5)], "Criação do estado") self.assertIn("2018-11-20", ap_holidays) self.assertEqual(ap_holidays[date(2018, 11, 20)], "Consciência Negra")
def check_holiday(self, *args): countries_holidays = holidays.CA(years=2021) countries_holidays += holidays.USA(years=2021) countries_holidays += holidays.BR(years=2021) countries_holidays += holidays.MW(years=2021) for e in args: dt = self.check_instance(e) if not dt: return False feriado = dt in countries_holidays return feriado
def test_TO_holidays(self): to_holidays = holidays.BR(state="TO") self.assertIn("2018-01-01", to_holidays) self.assertEqual(to_holidays[date(2018, 1, 1)], "Instalação de Tocantins, Ano novo") self.assertIn("2018-09-08", to_holidays) self.assertEqual(to_holidays[date(2018, 9, 8)], "Nossa Senhora da Natividade") self.assertIn("2018-10-05", to_holidays) self.assertEqual(to_holidays[date(2018, 10, 5)], "Criação de Tocantins")
def test_AL_holidays(self): al_holidays = holidays.BR(state="AL") self.assertIn("2018-06-24", al_holidays) self.assertEqual(al_holidays[date(2018, 6, 24)], "São João") self.assertIn("2018-06-29", al_holidays) self.assertEqual(al_holidays[date(2018, 6, 29)], "São Pedro") self.assertIn("2018-09-16", al_holidays) self.assertEqual(al_holidays[date(2018, 9, 16)], "Emancipação política de Alagoas") self.assertIn("2018-11-20", al_holidays) self.assertEqual(al_holidays[date(2018, 11, 20)], "Consciência Negra")
def test_AM_holidays(self): am_holidays = holidays.BR(state="AM") self.assertIn("2018-09-05", am_holidays) self.assertEqual( am_holidays[date(2018, 9, 5)], "Elevação do Amazonas à categoria de província", ) self.assertIn("2018-11-20", am_holidays) self.assertEqual(am_holidays[date(2018, 11, 20)], "Consciência Negra") self.assertIn("2018-12-08", am_holidays) self.assertEqual(am_holidays[date(2018, 12, 8)], "Dia de Nossa Senhora da Conceição")
def test_AC_holidays(self): ac_holidays = holidays.BR(state="AC") self.assertIn("2018-01-23", ac_holidays) self.assertEqual(ac_holidays[date(2018, 1, 23)], "Dia do evangélico") self.assertIn("2018-06-15", ac_holidays) self.assertEqual(ac_holidays[date(2018, 6, 15)], "Aniversário do Acre") self.assertIn("2018-09-05", ac_holidays) self.assertEqual(ac_holidays[date(2018, 9, 5)], "Dia da Amazônia") self.assertIn("2018-11-17", ac_holidays) self.assertEqual( ac_holidays[date(2018, 11, 17)], "Assinatura do Tratado de Petrópolis", )
def test_BR_holidays(self): self.holidays = holidays.BR(years=2018) self.assertIn("2018-01-01", self.holidays) self.assertEqual(self.holidays[date(2018, 1, 1)], "Ano novo") self.assertIn("2018-02-14", self.holidays) self.assertEqual( self.holidays[date(2018, 2, 14)], "Quarta-feira de cinzas (Início da Quaresma)", ) self.assertIn("2018-02-13", self.holidays) self.assertEqual(self.holidays[date(2018, 2, 13)], "Carnaval") self.assertIn("2018-03-30", self.holidays) self.assertEqual(self.holidays[date(2018, 3, 30)], "Sexta-feira Santa") self.assertIn("2018-02-13", self.holidays) self.assertEqual(self.holidays[date(2018, 2, 13)], "Carnaval") self.assertIn("2018-04-01", self.holidays) self.assertEqual(self.holidays[date(2018, 4, 1)], "Páscoa") self.assertIn("2018-04-21", self.holidays) self.assertEqual(self.holidays[date(2018, 4, 21)], "Tiradentes") self.assertIn("2018-05-01", self.holidays) self.assertEqual(self.holidays[date(2018, 5, 1)], "Dia Mundial do Trabalho") self.assertIn("2018-05-31", self.holidays) self.assertEqual(self.holidays[date(2018, 5, 31)], "Corpus Christi") self.assertIn("2018-09-07", self.holidays) self.assertEqual(self.holidays[date(2018, 9, 7)], "Independência do Brasil") self.assertIn("2018-10-12", self.holidays) self.assertEqual(self.holidays[date(2018, 10, 12)], "Nossa Senhora Aparecida") self.assertIn("2018-11-02", self.holidays) self.assertEqual(self.holidays[date(2018, 11, 2)], "Finados") self.assertIn("2018-11-15", self.holidays) self.assertEqual(self.holidays[date(2018, 11, 15)], "Proclamação da República") self.assertIn("2018-12-25", self.holidays) self.assertEqual(self.holidays[date(2018, 12, 25)], "Natal")
total = sum(w for c, w in choices) r = random.uniform(0, total) upto = 0 for c, w in choices: if upto + w >= r: return c, w upto += w return c, w ### # DATA GENERATION ### # get public holidays and length of the month public_holidays = holidays.BR(state=args.state, years=year) days_in_month = calendar.monthrange(year, month)[1] # check which days are valid, i.e. are specified workdays and not holidays valid_days = [] for day in range(args.fdom, min(days_in_month, ldom) + 1): date = datetime.date(year, month, day) if date not in public_holidays and date.weekday() in days_of_week: valid_days.append(day) # Distribute hours over valid days. Use exponential weights (after random shuffle) for days, # so some days are used often and some are used rarely. possible_days = valid_days random.shuffle(possible_days) weights = list(1 / np.arange(1, len(possible_days) + 1))
def test_GO_holidays(self): go_holidays = holidays.BR(state="GO") self.assertIn("2018-10-28", go_holidays) self.assertEqual(go_holidays[date(2018, 10, 28)], "Dia do Servidor Público")
def test_ES_holidays(self): es_holidays = holidays.BR(state="ES") self.assertIn("2018-10-28", es_holidays) self.assertEqual(es_holidays[date(2018, 10, 28)], "Dia do Servidor Público")
def test_MS_holidays(self): ms_holidays = holidays.BR(state="MS") self.assertIn("2018-10-11", ms_holidays) self.assertEqual(ms_holidays[date(2018, 10, 11)], "Criação do estado")
def test_MT_holidays(self): mt_holidays = holidays.BR(state="MT") self.assertIn("2018-11-20", mt_holidays) self.assertEqual(mt_holidays[date(2018, 11, 20)], "Consciência Negra")
def test_SE_holidays(self): se_holidays = holidays.BR(state="SE") self.assertIn("2018-07-08", se_holidays) self.assertEqual(se_holidays[date(2018, 7, 8)], "Autonomia política de Sergipe")
def test_MG_holidays(self): mg_holidays = holidays.BR(state="MG") self.assertIn("2018-04-21", mg_holidays) self.assertEqual(mg_holidays[date(2018, 4, 21)], "Data Magna de MG, Tiradentes")
def test_PB_holidays(self): pb_holidays = holidays.BR(state="PB") self.assertIn("2018-08-05", pb_holidays) self.assertEqual(pb_holidays[date(2018, 8, 5)], "Fundação do Estado")
def test_PR_holidays(self): pr_holidays = holidays.BR(state="PR") self.assertIn("2018-12-19", pr_holidays) self.assertEqual(pr_holidays[date(2018, 12, 19)], "Emancipação do Paraná")
def test_RR_holidays(self): rr_holidays = holidays.BR(state="RR") self.assertIn("2018-10-05", rr_holidays) self.assertEqual(rr_holidays[date(2018, 10, 5)], "Criação de Roraima")
def test_RO_holidays(self): ro_holidays = holidays.BR(state="RO") self.assertIn("2018-01-04", ro_holidays) self.assertEqual(ro_holidays[date(2018, 1, 4)], "Criação do estado") self.assertIn("2018-06-18", ro_holidays) self.assertEqual(ro_holidays[date(2018, 6, 18)], "Dia do Evangélico")
def test_RS_holidays(self): rs_holidays = holidays.BR(state="RS") self.assertIn("2018-09-20", rs_holidays) self.assertEqual(rs_holidays[date(2018, 9, 20)], "Revolução Farroupilha")
import vertica_python import verticapy import concurrent.futures from verticapy import * from verticapy.toolbox import * from verticapy.connections.connect import * from verticapy.learn.linear_model import LinearRegression from verticapy.learn.linear_model import ElasticNet from verticapy.learn.decomposition import PCA from pandas_datareader import data as web from datetime import datetime as dt #from finnhub import Client #finnhub_client = Client(api_key="bqu00jvrh5rb3gqqf9q0") ONE_DAY = datetime.timedelta(days=1) HOLIDAYS_BR = holidays.BR() logfile = 'vertica_service.log' conn_info = { 'host': 'localhost', 'port': 5433, 'user': '******', 'password': '******', 'database': 'stocks', 'use_prepared_statements': False } db_schema = "stocks" input_table = "daily_prices" key_columns = ["ts", "symbol"]
def test_CE_holidays(self): ce_holidays = holidays.BR(state="CE") self.assertIn("2018-03-19", ce_holidays) self.assertEqual(ce_holidays[date(2018, 3, 19)], "São José") self.assertIn("2018-03-25", ce_holidays) self.assertEqual(ce_holidays[date(2018, 3, 25)], "Data Magna do Ceará")