from unittest import TestCase

import darwin.sklearn_utils as dr
__author__ = 'oier'


class test_learners(TestCase):
    def test_get_clfmethod(self):
        met, gr = dr.get_clfmethod('DecisionTreeClassifier')
        assert( str(type(met)).find('DecisionTreeClassifier'))
        assert(gr['criterion'] == ['gini', 'entropy'] )

class test_selectors(TestCase):

    def test_get_selmethos(self):
        met, gr = dr.get_fsmethod('RFE')
        assert( str(type(met)).find('RFE'))
        assert(gr['step'] == [0.01, 0.05, 0.1])
3cbc62656bebb91061f6b4a2f8c804d28097825b
<mask token>


class test_learners(TestCase):
    <mask token>


class test_selectors(TestCase):

    def test_get_selmethos(self):
        met, gr = dr.get_fsmethod('RFE')
        assert str(type(met)).find('RFE')
        assert gr['step'] == [0.01, 0.05, 0.1]

<mask token>


class test_learners(TestCase):

    def test_get_clfmethod(self):
        met, gr = dr.get_clfmethod('DecisionTreeClassifier')
        assert str(type(met)).find('DecisionTreeClassifier')
        assert gr['criterion'] == ['gini', 'entropy']


class test_selectors(TestCase):

    def test_get_selmethos(self):
        met, gr = dr.get_fsmethod('RFE')
        assert str(type(met)).find('RFE')
        assert gr['step'] == [0.01, 0.05, 0.1]

<mask token>
__author__ = 'oier'


class test_learners(TestCase):

    def test_get_clfmethod(self):
        met, gr = dr.get_clfmethod('DecisionTreeClassifier')
        assert str(type(met)).find('DecisionTreeClassifier')
        assert gr['criterion'] == ['gini', 'entropy']


class test_selectors(TestCase):

    def test_get_selmethos(self):
        met, gr = dr.get_fsmethod('RFE')
        assert str(type(met)).find('RFE')
        assert gr['step'] == [0.01, 0.05, 0.1]

from unittest import TestCase
import darwin.sklearn_utils as dr
__author__ = 'oier'


class test_learners(TestCase):

    def test_get_clfmethod(self):
        met, gr = dr.get_clfmethod('DecisionTreeClassifier')
        assert str(type(met)).find('DecisionTreeClassifier')
        assert gr['criterion'] == ['gini', 'entropy']


class test_selectors(TestCase):

    def test_get_selmethos(self):
        met, gr = dr.get_fsmethod('RFE')
        assert str(type(met)).find('RFE')
        assert gr['step'] == [0.01, 0.05, 0.1]

from unittest import TestCase

import darwin.sklearn_utils as dr
__author__ = 'oier'


class test_learners(TestCase):
    def test_get_clfmethod(self):
        met, gr = dr.get_clfmethod('DecisionTreeClassifier')
        assert( str(type(met)).find('DecisionTreeClassifier'))
        assert(gr['criterion'] == ['gini', 'entropy'] )

class test_selectors(TestCase):

    def test_get_selmethos(self):
        met, gr = dr.get_fsmethod('RFE')
        assert( str(type(met)).find('RFE'))
        assert(gr['step'] == [0.01, 0.05, 0.1])