def test_fail3(self):
with pytest.raises(AssertionError):
stata_merge(self.df1,
self.df1_part,
on='a',
how='left',
assertval=3)
def test_assert3(self):
result = stata_merge(self.df1, self.df1_match, on='a', how='left',
assertval=3)
expected = pd.DataFrame(
np.array([[0, 1, 2], [1, 1, 1], [11, 11, 11]]).T,
columns=['a', 'b', 'c'],
dtype=np.float64)
assert_frame_equal(result, expected)
def test_part_left(self):
result = stata_merge(self.df1, self.df1_part, on='a', how='left')
expected = pd.DataFrame(
np.array([[0, 1, 2], [1, 1, 1], [9, 10, np.nan], [3, 3, 1]]).T,
columns=['a', 'b', 'c', '_m'],
dtype=np.float64)
print(result)
print(expected)
assert_frame_equal(result, expected)
def test_part_left(self):
result = stata_merge(self.df1, self.df1_part, on='a', how='left')
expected = pd.DataFrame(np.array([[0, 1, 2], [1, 1, 1],
[9, 10, np.nan], [3, 3, 1]]).T,
columns=['a', 'b', 'c', '_m'],
dtype=np.float64)
print(result)
print(expected)
assert_frame_equal(result, expected)
def test_autonaming(self):
left = self.df1.rename(columns={'a': 'tmpa'})
right = self.df1_match.rename(columns={'a': 'tmpb'})
result = stata_merge(left, right, left_on='tmpa', right_on='tmpb',
how='left', assertval=3)
expected = pd.DataFrame(
np.array([[0, 1, 2], [1, 1, 1], [0, 1, 2], [11, 11, 11]]).T,
columns=['tmpa', 'b', 'tmpb', 'c'],
dtype=np.float64)
assert_frame_equal(result, expected)
def test_assert3(self):
result = stata_merge(self.df1,
self.df1_match,
on='a',
how='left',
assertval=3)
expected = pd.DataFrame(np.array([[0, 1, 2], [1, 1, 1], [11, 11,
11]]).T,
columns=['a', 'b', 'c'],
dtype=np.float64)
assert_frame_equal(result, expected)
def test_autonaming(self):
left = self.df1.rename(columns={'a': 'tmpa'})
right = self.df1_match.rename(columns={'a': 'tmpb'})
result = stata_merge(left,
right,
left_on='tmpa',
right_on='tmpb',
how='left',
assertval=3)
expected = pd.DataFrame(np.array([[0, 1, 2], [1, 1, 1], [0, 1, 2],
[11, 11, 11]]).T,
columns=['tmpa', 'b', 'tmpb', 'c'],
dtype=np.float64)
assert_frame_equal(result, expected)
def test_fail3(self):
with pytest.raises(AssertionError):
stata_merge(self.df1, self.df1_part, on='a', how='left',
assertval=3)
def test_fail3(self):
stata_merge(self.df1, self.df1_part, on='a', how='left', assertval=3)