#==================================================================================================================================================== 

# AULA 03
# MATERIA: ECONOMETRIA II
# PROFESSOR: DANIEL DOMINGUES DOS SANTOS
# MONITOR: DIEGO MENEZES
# R - VERSAO UTILIZADA: 3.4.0 (2017-04-21)

# SCRIPT: exercicio pratico -- experimento aleatorio
# ETAPA: setup

#====================================================================================================================================================

# construcao da base
n <- 6000
base <- data.frame(rbinom(n, 1, 0.5))
names(base)[1] <- "treat"

# atribuicao das características individuais
gender <- rbinom(n, 1, 0.5)
age    <- rbinom(n, 1, 0.7)
study  <- rbinom(n, 1, 0.4)

base <- cbind(base, gender, age, study)

# check dos primeiros 15 individuos da amostra
head(base, n = 15)

# divisao entre grupos de tratamento e controle
tratados <- base[base$treat == 1,]
controle <- base[base$treat == 0,]

nt <- nrow(tratados)
nc <- nrow(controle)

# atribuicao das variaveis de resultado
t_em   <- rbinom(nt, 1, prob = (0.9))
t_enem <- rbinom(nt, 1, prob = (0.8))
c_em   <- rbinom(nc, 1, 0.7)
c_enem <- rbinom(nc, 1, 0.5)

tratados = cbind(tratados, t_em, t_enem)
controle = cbind(controle, c_em, c_enem)

names(tratados)[5] = "em"
names(tratados)[6] = "enem"
names(controle)[5] = "em"
names(controle)[6] = "enem"

base <- rbind(tratados, controle)

# check dos primeiros 15 individuos da amostra
head(base, n = 15)


# END OF SCRIPT -------------------------------------------------------------------------------------------------------------------------------------