search

m3dev / gokart

Gokart solves reproducibility, task dependencies, constraints of good code, and ease of use for Machine Learning Pipeline.
https://gokart.readthedocs.io/en/latest/
MIT License
318 stars 57 forks source link

Feature Request: make mock task for test #245

Open vaaaaanquish opened 3 years ago

vaaaaanquish commented 3 years ago

I want to be able to create dependent tasks easily for regression test.

image:

X = gokart.MockTask(dump_value='hoge')

class Example(gokart.TaskOnKart):
    target = gokart.TaskInstanceParameter()

    ...

task = Example(target=X)
df = gokart.build(task)
pd.testing.assert_frame_equal(df, expected)