Skip to content

Commit

Permalink
utility function to create sample df
Browse files Browse the repository at this point in the history
  • Loading branch information
ClimbsRocks committed Dec 3, 2017
1 parent b4351ba commit 61523d1
Showing 1 changed file with 14 additions and 0 deletions.
14 changes: 14 additions & 0 deletions tests/utils_testing.py
Original file line number Diff line number Diff line change
Expand Up @@ -107,3 +107,17 @@ def train_basic_multilabel_classifier(df_twitter_train):
ml_predictor.train(df_twitter_train)

return ml_predictor


import pandas as pd
import datetime
def make_test_df():
today = datetime.datetime.today()
raw_input = {
'a': [1,2,3,4,5]
, 'b': [6,7,8,9,10]
, 'text_col': ['hi', 'there', 'mesmerizingly', 'intriguing', 'world']
, 'date_col': [today, today - datetime.timedelta(days=1), today - datetime.timedelta(days=2), today - datetime.timedelta(days=3), today - datetime.timedelta(days=4)]
}
df = pd.DataFrame(raw_input)
return df

0 comments on commit 61523d1

Please sign in to comment.