versatiles_container 4.10.0

A toolbox for converting, checking and serving map tiles in various formats.
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

//! Shared test utilities and fixtures for the versatiles_container crate.
//!
//! This module provides common test helpers that can be reused across test modules,
//! including downstream crates that pull `versatiles_container` with the `test`
//! feature enabled.

use anyhow::Result;
use versatiles_core::TileBBox;
use versatiles_image::{DynamicImage, ImageBuffer};

use crate::TileSource;

/// Creates a tiny 2x2 RGB test image with known pixel values.
///
/// Pixel layout:
/// - (0,0): red (255, 0, 0)
/// - (1,0): green (0, 255, 0)
/// - (0,1): blue (0, 0, 255)
/// - (1,1): misc (10, 20, 30)
#[must_use]
pub fn tiny_rgb_image() -> DynamicImage {
	let data = vec![
		255, 0, 0, // red
		0, 255, 0, // green
		0, 0, 255, // blue
		10, 20, 30, // misc
	];
	DynamicImage::ImageRgb8(ImageBuffer::from_vec(2, 2, data).expect("vec->img"))
}

/// Asserts that a source's `tile_coord_stream` and `tile_stream` produce the
/// same number of items for the given `bbox`.
///
/// This invariant is required by the [`TileSource`] trait:
/// any implementation that diverges between the two streams will silently
/// break the converter's progress accounting (and any other code that counts
/// via the lighter `tile_coord_stream`). See the trait docs for the rationale.
///
/// # Panics
/// Panics with a useful message when the two counts disagree.
pub async fn assert_stream_counts_agree(source: &dyn TileSource, bbox: TileBBox) -> Result<()> {
	let coord_count = source.tile_coord_stream(bbox).await?.drain_and_count().await;
	let tile_count = source.tile_stream(bbox).await?.to_vec().await.len() as u64;
	assert_eq!(
		coord_count, tile_count,
		"TileSource invariant violated for bbox {bbox:?}: tile_coord_stream yielded {coord_count}, tile_stream yielded {tile_count}",
	);
	Ok(())
}

#[cfg(test)]
mod tests {
	use versatiles_image::GenericImageView;

	use super::*;

	#[test]
	fn tiny_rgb_image_has_expected_dimensions() {
		let img = tiny_rgb_image();
		assert_eq!(img.dimensions(), (2, 2));
	}

	#[test]
	fn tiny_rgb_image_has_expected_pixels() {
		let img = tiny_rgb_image();
		let p00 = img.get_pixel(0, 0);
		let p10 = img.get_pixel(1, 0);
		let p01 = img.get_pixel(0, 1);
		let p11 = img.get_pixel(1, 1);

		assert_eq!(p00.0[0..3], [255, 0, 0]); // red
		assert_eq!(p10.0[0..3], [0, 255, 0]); // green
		assert_eq!(p01.0[0..3], [0, 0, 255]); // blue
		assert_eq!(p11.0[0..3], [10, 20, 30]); // misc
	}
}