aipack 0.8.23

Command Agent runner to accelerate production coding with genai.
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

type Result<T> = core::result::Result<T, Box<dyn std::error::Error>>;

use super::*;
use crate::support::AsStrsExt as _;
use simple_fs::SPath;

#[test]
fn test_support_csv_load_headers_simple() -> Result<()> {
	// -- Setup & Fixtures
	let path = SPath::new("tests-data/sandbox-01/example.csv");

	// -- Exec
	let headers = load_csv_headers(&path, None)?;

	// -- Check
	let expected = vec!["id", "name", "email"];
	assert_eq!(headers.x_as_strs(), expected);

	Ok(())
}

#[test]
fn test_support_csv_to_csv_row_simple() -> Result<()> {
	let row = values_to_csv_row(&["a".to_string(), "b,c".to_string(), "d".to_string()], None)?;
	assert_eq!(row, r#"a,"b,c",d"#);
	Ok(())
}

#[test]
fn test_support_csv_to_csv_row_custom_delimiter() -> Result<()> {
	let options = CsvOptions {
		delimiter: Some(";".to_string()),
		..Default::default()
	};
	let row = values_to_csv_row(&["a".to_string(), "b;c".to_string(), "d".to_string()], Some(options))?;
	assert_eq!(row, r#"a;"b;c";d"#);
	Ok(())
}

#[test]
fn test_support_csv_load_simple() -> Result<()> {
	// -- Setup & Fixtures
	let path = SPath::new("tests-data/sandbox-01/example.csv");

	// -- Exec
	let res = load_csv(&path, None)?;

	// -- Check
	let expected_headers = vec!["id", "name", "email"];
	assert_eq!(res.headers.x_as_strs(), expected_headers);

	let expected_content = vec![vec!["1", "Alice", "alice@example.com"], vec!["2", "Bob", "bob@example.com"]];
	let content_as_strs: Vec<Vec<&str>> = res.rows.iter().map(|row| row.x_as_strs()).collect();
	assert_eq!(content_as_strs, expected_content);

	Ok(())
}

#[test]
fn test_support_csv_load_no_header() -> Result<()> {
	// -- Setup & Fixtures
	let path = SPath::new("tests-data/sandbox-01/example.csv");

	// -- Exec
	let res = load_csv(
		&path,
		Some(CsvOptions {
			has_header: Some(false),
			..Default::default()
		}),
	)?;

	// -- Check
	assert!(res.headers.is_empty(), "Headers should be empty when no_header is true");

	let expected_content = vec![
		vec!["id", "name", "email"],
		vec!["1", "Alice", "alice@example.com"],
		vec!["2", "Bob", "bob@example.com"],
	];

	let content_as_strs: Vec<Vec<&str>> = res.rows.iter().map(|row| row.x_as_strs()).collect();
	assert_eq!(content_as_strs, expected_content);

	Ok(())
}

#[test]
fn test_support_csv_load_with_header_labels() -> Result<()> {
	// -- Setup & Fixtures
	let content = "ID Code,Full Name,Contact Email\n1,Alice,alice@x.com";
	let mut labels = HashMap::new();
	labels.insert("id".to_string(), "ID Code".to_string());
	labels.insert("name".to_string(), "Full Name".to_string());
	// "Contact Email" not mapped, should remain as is.

	let options = CsvOptions {
		header_labels: Some(labels),
		..Default::default()
	};

	// -- Exec
	let res = parse_csv(content, Some(options))?;

	// -- Check
	let expected_headers = vec!["id", "name", "Contact Email"];
	assert_eq!(res.headers, expected_headers);
	assert_eq!(res.rows.len(), 1);
	assert_eq!(res.rows[0], vec!["1", "Alice", "alice@x.com"]);

	Ok(())
}

#[test]
fn test_support_csv_load_headers_remapped() -> Result<()> {
	// -- Setup & Fixtures
	// Content: id,name,email
	let path = SPath::new("tests-data/sandbox-01/example.csv");

	let mut labels = HashMap::new();
	labels.insert("ID".to_string(), "id".to_string()); // map "id" -> "ID"
	// "name" and "email" unmapped

	let options = CsvOptions {
		header_labels: Some(labels),
		..Default::default()
	};

	// -- Exec
	let headers = load_csv_headers(&path, Some(options))?;

	// -- Check
	// Original: ["id", "name", "email"]
	// Remapped: ["ID", "name", "email"]
	let expected = vec!["ID", "name", "email"];
	assert_eq!(headers, expected);

	Ok(())
}