ml-cellar 0.2.0

CLI of ML model registry for minimum MLOps
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144

use assert_cmd::cargo::cargo_bin_cmd;

/// This test uses `tests/fixtures/test_registry/test_check_for_files/0.1`
/// This directory contains all required files.
/// Expected to success for check command.
#[test]
fn test_check_required_files() {
    let test_dir = "tests/fixtures/test_registry/test_check_for_files/0.1";
    let mut cmd = cargo_bin_cmd!("ml-cellar");
    cmd.arg("check").arg(test_dir);
    cmd.assert().success();
}

/// This test uses `tests/fixtures/test_registry/test_check_for_files/0.2`
/// This directory lacks `config.yaml`, which is required file.
/// Expected to failure for check command.
#[test]
fn test_check_missing_required_files() {
    let test_dir = "tests/fixtures/test_registry/test_check_for_files/0.2";
    let mut cmd = cargo_bin_cmd!("ml-cellar");
    cmd.arg("check").arg(test_dir);
    cmd.assert().failure();
}

/// This test uses `tests/fixtures/test_registry/test_check_for_files/0.3`
/// This directory contains all required files and optional files (config.py).
/// Expected to success for check command because config.toml defines any kind of files as optional files.
#[test]
fn test_check_optional_files() {
    let test_dir = "tests/fixtures/test_registry/test_check_for_files/0.3";
    let mut cmd = cargo_bin_cmd!("ml-cellar");
    cmd.arg("check").arg(test_dir);
    cmd.assert().success();
}

/// This test uses `tests/fixtures/test_registry/test_check_for_project_files/0.1`
/// This directory contains all required files in project's directory (base/).
/// Expected to success for check command.
#[test]
fn test_check_required_files_for_project() {
    let test_dir = "tests/fixtures/test_registry/test_check_for_project_files/base/0.1";
    let mut cmd = cargo_bin_cmd!("ml-cellar");
    cmd.arg("check").arg(test_dir);
    cmd.assert().success();
}

/// This test uses `tests/fixtures/test_registry/test_check_for_project_files/0.2`
/// This directory lacks `config.yaml`, which is required file.
/// Expected to failure for check command.
#[test]
fn test_check_missing_required_files_for_project() {
    let test_dir = "tests/fixtures/test_registry/test_check_for_project_files/base/0.2";
    let mut cmd = cargo_bin_cmd!("ml-cellar");
    cmd.arg("check").arg(test_dir);
    cmd.assert().failure();
}

/// This test uses `tests/fixtures/test_registry/test_check_for_project_files/0.3`
/// This directory contains all required files and `config.py`, which is not defined as optional files.
/// Expected to failure for check command.
#[test]
fn test_check_optional_files_for_project() {
    let test_dir = "tests/fixtures/test_registry/test_check_for_project_files/base/0.3";
    let mut cmd = cargo_bin_cmd!("ml-cellar");
    cmd.arg("check").arg(test_dir);
    cmd.assert().failure();
}

/// This test uses `tests/fixtures/test_registry/test_check_for_version/base/1` as test directory
/// The version "1" is wrong and should be 1.0 as X.Y.
/// Expected to failure for check command.
#[test]
fn test_check_version_x_y_missing_to_x() {
    let test_dir = "tests/fixtures/test_registry/test_check_for_version/1";
    let mut cmd = cargo_bin_cmd!("ml-cellar");
    cmd.arg("check").arg(test_dir);
    cmd.assert().failure();
}

/// This test uses `tests/fixtures/test_registry/test_check_for_version/base/1.0.0` as test directory
/// The version "1.0.0" is wrong and should be 1.0 as X.Y.
/// Expected to failure for check command.
#[test]
fn test_check_version_x_y_missing_to_x_y_z() {
    let test_dir = "tests/fixtures/test_registry/test_check_for_version/base/1.0.0";
    let mut cmd = cargo_bin_cmd!("ml-cellar");
    cmd.arg("check").arg(test_dir);
    cmd.assert().failure();
}

/// This test uses `tests/fixtures/test_registry/test_check_for_version/projectA/0.1` as test directory.
/// The version "0.1" is wrong and should be 0.1.0 as X.Y.Z.
/// Expected to failure for check command.
#[test]
fn test_check_version_x_y_z_missing_to_x_y() {
    let test_dir = "tests/fixtures/test_registry/test_check_for_version/projectA/0.1";
    let mut cmd = cargo_bin_cmd!("ml-cellar");
    cmd.arg("check").arg(test_dir);
    cmd.assert().failure();
}

/// This test uses `tests/fixtures/test_registry/test_check_for_version/projectA/0.1.1` as test directory.
/// The version "0.1.1" is correct versioning as X.Y.Z.
/// Expected to success for check command.
#[test]
fn test_check_version_x_y_z() {
    let test_dir = "tests/fixtures/test_registry/test_check_for_version/projectA/0.1.1";
    let mut cmd = cargo_bin_cmd!("ml-cellar");
    cmd.arg("check").arg(test_dir);
    cmd.assert().success();
}

/// This test uses `tests/fixtures/test_registry/test_check_for_version/pretrained/20250101` as test directory.
/// The version "20250101" is correct versioning as YYYYMMDD.
/// Expected to success for check command.
#[test]
fn test_check_version_for_yyyymmdd() {
    let test_dir = "tests/fixtures/test_registry/test_check_for_version/pretrained/20250101";
    let mut cmd = cargo_bin_cmd!("ml-cellar");
    cmd.arg("check").arg(test_dir);
    cmd.assert().success();
}

/// This test uses `tests/fixtures/test_registry/test_check_for_version/pretrained/0.1` as test directory
/// The version "0.1" is wrong and should be 20250101 as YYYYMMDD.
/// Expected to failure for check command.
#[test]
fn test_check_version_yyyymmdd_missing_to_x_y() {
    let test_dir = "tests/fixtures/test_registry/test_check_for_version/pretrained/0.1";
    let mut cmd = cargo_bin_cmd!("ml-cellar");
    cmd.arg("check").arg(test_dir);
    cmd.assert().failure();
}

/// This test uses `tests/fixtures/test_registry/test_check_for_version/pretrained/202501030` as test directory.
/// The version "202501030" is wrong for YYYYMMDD.
/// Expected to failure for check command.
#[test]
fn test_check_version_for_missing_yyyymmdd() {
    let test_dir = "tests/fixtures/test_registry/test_check_for_version/pretrained/202501030";
    let mut cmd = cargo_bin_cmd!("ml-cellar");
    cmd.arg("check").arg(test_dir);
    cmd.assert().failure();
}