class TestDID(TestAPI):
def setUp(self):
SIZE = 2000
assignment = np.random.binomial(1,0.5, size=SIZE)
pre_experiment = assignment + np.random.normal(-1, size=SIZE)
start = assignment + np.random.normal(1, size=SIZE)
end = start + np.random.normal(2.*assignment) + np.random.normal(2, size=SIZE)
self.X_pre = pd.DataFrame({'Start' : pre_experiment, 'End' : start, 'assignment' : assignment})
self.X = pd.DataFrame({'Start' : start, 'End' : end, 'assignment' : assignment})
self.did = DifferenceInDifferences()
def test_assumption_tester(self):
assert self.did.test_parallel_trend(self.X_pre)
self.X_pre['End'] += self.X_pre['assignment']
assert not self.did.test_parallel_trend(self.X_pre)
def test_did_estimator(self):
lower, exp, upper = self.did.average_treatment_effect(self.X)
assert 1.8 <= exp <= 2.2
assert lower <= exp <= upper
self.did = DifferenceInDifferences(robust=True)
lower, exp, upper = self.did.average_treatment_effect(self.X)
assert 1.8 <= exp <= 2.2
assert lower <= exp <= upper
class TestDID(TestAPI):
def setUp(self):
SIZE = 2000
assignment = np.random.binomial(1, 0.5, size=SIZE)
pre_experiment = assignment + np.random.normal(-1, size=SIZE)
start = assignment + np.random.normal(1, size=SIZE)
end = start + np.random.normal(2. * assignment) + np.random.normal(
2, size=SIZE)
self.X_pre = pd.DataFrame({
'Start': pre_experiment,
'End': start,
'assignment': assignment
})
self.X = pd.DataFrame({
'Start': start,
'End': end,
'assignment': assignment
})
self.did = DifferenceInDifferences()
def test_assumption_tester(self):
assert self.did.test_parallel_trend(self.X_pre)
self.X_pre['End'] += self.X_pre['assignment']
assert not self.did.test_parallel_trend(self.X_pre)
def test_did_estimator(self):
lower, exp, upper = self.did.average_treatment_effect(self.X)
assert 1.8 <= exp <= 2.2
assert lower <= exp <= upper
self.did = DifferenceInDifferences(robust=True)
lower, exp, upper = self.did.average_treatment_effect(self.X)
assert 1.8 <= exp <= 2.2
assert lower <= exp <= upper
def test_did_estimator(self):
lower, exp, upper = self.did.average_treatment_effect(self.X)
assert 1.8 <= exp <= 2.2
assert lower <= exp <= upper
self.did = DifferenceInDifferences(robust=True)
lower, exp, upper = self.did.average_treatment_effect(self.X)
assert 1.8 <= exp <= 2.2
assert lower <= exp <= upper
def test_did_estimator(self):
lower, exp, upper = self.did.average_treatment_effect(self.X)
assert 1.8 <= exp <= 2.2
assert lower <= exp <= upper
self.did = DifferenceInDifferences(robust=True)
lower, exp, upper = self.did.average_treatment_effect(self.X)
assert 1.8 <= exp <= 2.2
assert lower <= exp <= upper
def setUp(self):
SIZE = 2000
assignment = np.random.binomial(1,0.5, size=SIZE)
pre_experiment = assignment + np.random.normal(-1, size=SIZE)
start = assignment + np.random.normal(1, size=SIZE)
end = start + np.random.normal(2.*assignment) + np.random.normal(2, size=SIZE)
self.X_pre = pd.DataFrame({'Start' : pre_experiment, 'End' : start, 'assignment' : assignment})
self.X = pd.DataFrame({'Start' : start, 'End' : end, 'assignment' : assignment})
self.did = DifferenceInDifferences()
def setUp(self):
SIZE = 2000
assignment = np.random.binomial(1, 0.5, size=SIZE)
pre_experiment = assignment + np.random.normal(-1, size=SIZE)
start = assignment + np.random.normal(1, size=SIZE)
end = start + np.random.normal(2. * assignment) + np.random.normal(
2, size=SIZE)
self.X_pre = pd.DataFrame({
'Start': pre_experiment,
'End': start,
'assignment': assignment
})
self.X = pd.DataFrame({
'Start': start,
'End': end,
'assignment': assignment
})
self.did = DifferenceInDifferences()