def test_only_complete(self):
result = h.pb_only_complete_drafts(self.df)
tm.assert_frame_equal(result, self.df)
bad = self.df.copy()
bad.loc[0, 'team_id'] = np.nan
bad['match_id'] = bad['match_id'] + 1
bad = pd.concat([bad, self.df])
result = h.pb_only_complete_drafts(bad)
tm.assert_frame_equal(result, self.df, check_dtype=False)
def test_full_draft(self):
bad = self.df.copy().loc[:5]
bad['match_id'] = bad['match_id'] + 1
bad = pd.concat([bad, self.df])
result = h.pb_only_complete_drafts(bad)
tm.assert_frame_equal(result, self.df, check_dtype=False)