-
Notifications
You must be signed in to change notification settings - Fork 9
Expand file tree
/
Copy pathtest_image.rs
More file actions
129 lines (113 loc) · 3.73 KB
/
test_image.rs
File metadata and controls
129 lines (113 loc) · 3.73 KB
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
use plotpy::{Image, Plot, StrError};
use std::fs::File;
use std::io::{BufRead, BufReader};
use std::path::Path;
const OUT_DIR: &str = "/tmp/plotpy/integ_tests";
#[test]
fn test_image_1() -> Result<(), StrError> {
// data
let data = [
[0.8, 2.4, 2.5, 3.9, 0.0, 4.0, 0.0],
[2.4, 0.0, 4.0, 1.0, 2.7, 0.0, 0.0],
[1.1, 2.4, 0.8, 4.3, 1.9, 4.4, 0.0],
[0.6, 0.0, 0.3, 0.0, 3.1, 0.0, 0.0],
[0.7, 1.7, 0.6, 2.6, 2.2, 6.2, 0.0],
[1.3, 1.2, 0.0, 0.0, 0.0, 3.2, 5.1],
[0.1, 2.0, 0.0, 1.4, 0.0, 1.9, 6.3],
];
// image plot and options
let mut img = Image::new();
img.set_colormap_name("terrain").set_extra("alpha=0.8").draw(&data);
let mut plot = Plot::new();
plot.add(&img);
// save figure
let path = Path::new(OUT_DIR).join("integ_image_1.svg");
plot.set_show_errors(true).save(&path)?;
// check number of lines
let file = File::open(path).map_err(|_| "cannot open file")?;
let buffered = BufReader::new(file);
let lines_iter = buffered.lines();
let c = lines_iter.count();
assert!(c > 420 && c < 500);
Ok(())
}
#[test]
fn test_image_with_rgb() -> Result<(), StrError> {
let data = vec![
// --- Row 0 ---
vec![
vec![1.0, 0.0, 0.0], // Pixel 0,0: Red
vec![0.0, 1.0, 0.0], // Pixel 0,1: Green
vec![0.0, 0.0, 1.0], // Pixel 0,2: Blue
],
// --- Row 1 ---
vec![
vec![1.0, 1.0, 0.0], // Pixel 1,0: Yellow
vec![1.0, 0.0, 1.0], // Pixel 1,1: Magenta
vec![0.0, 1.0, 1.0], // Pixel 1,2: Cyan
],
// --- Row 2 ---
vec![
vec![0.5, 0.5, 0.5], // Pixel 2,0: Gray
vec![1.0, 1.0, 1.0], // Pixel 2,1: White
vec![0.0, 0.0, 0.0], // Pixel 2,2: Black
],
];
// image plot and options
let mut img = Image::new();
img.set_colormap_name("terrain")
.set_extra("alpha=0.8")
.draw_rgb_or_rgba(&data);
let mut plot = Plot::new();
plot.add(&img);
// save figure
let path = Path::new(OUT_DIR).join("integ_image_with_rgb.svg");
plot.set_show_errors(true).save(&path)?;
// check number of lines
let file = File::open(path).map_err(|_| "cannot open file")?;
let buffered = BufReader::new(file);
let lines_iter = buffered.lines();
let c = lines_iter.count();
assert!(c > 400 && c < 430);
Ok(())
}
#[test]
fn test_image_with_rgba() -> Result<(), StrError> {
let data = vec![
// --- Row 0 ---
vec![
vec![1.0, 0.0, 0.0, 0.5], // Pixel 0,0: Red
vec![0.0, 1.0, 0.0, 0.5], // Pixel 0,1: Green
vec![0.0, 0.0, 1.0, 0.5], // Pixel 0,2: Blue
],
// --- Row 1 ---
vec![
vec![1.0, 1.0, 0.0, 0.8], // Pixel 1,0: Yellow
vec![1.0, 0.0, 1.0, 0.8], // Pixel 1,1: Magenta
vec![0.0, 1.0, 1.0, 0.8], // Pixel 1,2: Cyan
],
// --- Row 2 ---
vec![
vec![0.5, 0.5, 0.5, 0.2], // Pixel 2,0: Gray
vec![1.0, 1.0, 1.0, 0.2], // Pixel 2,1: White
vec![0.0, 0.0, 0.0, 0.2], // Pixel 2,2: Black
],
];
// image plot and options
let mut img = Image::new();
img.set_colormap_name("terrain")
.set_extra("alpha=0.8")
.draw_rgb_or_rgba(&data);
let mut plot = Plot::new();
plot.add(&img);
// save figure
let path = Path::new(OUT_DIR).join("integ_image_with_rgba.svg");
plot.set_show_errors(true).save(&path)?;
// check number of lines
let file = File::open(path).map_err(|_| "cannot open file")?;
let buffered = BufReader::new(file);
let lines_iter = buffered.lines();
let c = lines_iter.count();
assert!(c > 400 && c < 430);
Ok(())
}