wehub-resource-sync
55 lines
1.4 KiB
Python
55 lines
1.4 KiB
Python
import unittest
|
|
|
|
|
|
class CoSTEERTest(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.test_competition = "aerial-cactus-identification"
|
|
|
|
def tearDown(self):
|
|
pass
|
|
|
|
def to_str(self, obj):
|
|
return "".join(str(obj).split())
|
|
|
|
def test_data_loader(self):
|
|
from rdagent.components.coder.data_science.raw_data_loader.test import (
|
|
develop_one_competition,
|
|
)
|
|
|
|
# if all tasks in exp are failed, will raise CoderError
|
|
exp = develop_one_competition(self.test_competition)
|
|
|
|
def test_feature(self):
|
|
from rdagent.components.coder.data_science.feature.test import (
|
|
develop_one_competition,
|
|
)
|
|
|
|
exp = develop_one_competition(self.test_competition)
|
|
|
|
def test_model(self):
|
|
from rdagent.components.coder.data_science.model.test import (
|
|
develop_one_competition,
|
|
)
|
|
|
|
exp = develop_one_competition(self.test_competition)
|
|
|
|
def test_ensemble(self):
|
|
from rdagent.components.coder.data_science.ensemble.test import (
|
|
develop_one_competition,
|
|
)
|
|
|
|
exp = develop_one_competition(self.test_competition)
|
|
|
|
def test_workflow(self):
|
|
from rdagent.components.coder.data_science.workflow.test import (
|
|
develop_one_competition,
|
|
)
|
|
|
|
exp = develop_one_competition(self.test_competition)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|
|
# pytest test/utils/coder/test_CoSTEER.py
|