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)