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"))


if __name__ == "__main__":
    unittest.main()