def test_gini_default(self, idadf):
if len(idadf.columns) > 1:
result = gini(idadf, features = idadf.columns)
assert(isinstance(result, pandas.core.series.Series))
assert(len(result.index) == len(idadf.columns))
result2 = gini(idadf)
assert(all(result == result2))
def test_gini_default(self, idadf):
if len(idadf.columns) > 1:
result = gini(idadf, features=idadf.columns)
assert (isinstance(result, pandas.core.series.Series))
assert (len(result.index) == len(idadf.columns))
result2 = gini(idadf)
assert (all(result == result2))
def test_gini_one_column(self, idadf):
if len(idadf.columns) >= 1:
result = gini(idadf, features = idadf.columns[0])
assert(isinstance(result, float))
def test_gini_multiple_columns(self, idadf):
if len(idadf.columns) > 1:
result = gini(idadf, features = [idadf.columns[0],idadf.columns[1]])
assert(isinstance(result, pandas.core.series.Series))
assert(len(result) == 2)
def test_gini_one_column(self, idadf):
if len(idadf.columns) >= 1:
result = gini(idadf, features=idadf.columns[0])
assert (isinstance(result, float))
def test_gini_multiple_columns(self, idadf):
if len(idadf.columns) > 1:
result = gini(idadf, features=[idadf.columns[0], idadf.columns[1]])
assert (isinstance(result, pandas.core.series.Series))
assert (len(result) == 2)
