text_basic.py

import pandas as pd 
from core.train import train_model

def text_train_small():
    df = pd.read_csv("data/processed/mvp_100_listed.csv")
    models, expl, base, auc, feats = train_model(df)
    assert auc > 0.5
    assert len(feats) >=10