24 lines
760 B
Python
24 lines
760 B
Python
import unittest
|
|
from pathlib import Path
|
|
|
|
from cbpred.__main__ import build_parser
|
|
|
|
|
|
class ArtifactPathTests(unittest.TestCase):
|
|
def test_cli_defaults_use_versioned_artifact_directory(self):
|
|
parser = build_parser()
|
|
|
|
train = parser.parse_args(["train"])
|
|
run = parser.parse_args(["run"])
|
|
predict = parser.parse_args(["predict", "--csv", "data.csv"])
|
|
|
|
self.assertEqual(train.artifact_dir, Path("artifacts/v1"))
|
|
self.assertEqual(run.artifact_dir, Path("artifacts/v1"))
|
|
self.assertEqual(predict.artifact, Path("artifacts/v1/model.joblib"))
|
|
daily = parser.parse_args(["prepare", "--timeframe", "1d"])
|
|
self.assertEqual(daily.timeframe, "1d")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|