anndata-rs/anndata-test-utils/tests/tests.rs at 3c8f3347191b1a7eca99425c71cedf781dd5b55c · ilan-gold/anndata-rs · GitHub

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
use anndata_test_utils as utils;
use anndata_test_utils::with_tmp_dir;
use anndata_hdf5::H5;
use anndata_zarr::Zarr;
use anndata::{AnnData, Backend};

#[test]
fn test_basic() {
    utils::test_basic::<H5>();
    utils::test_basic::<Zarr>();
}

#[test]
fn test_complex_dataframe() {
    let input = "tests/data/sample.h5ad";
    with_tmp_dir(|dir| {
        let file = dir.join("test.h5");
        let adata = AnnData::<H5>::open(H5::open(&input).unwrap()).unwrap();
        adata.write::<H5, _>(file, None, None).unwrap();

        let file = dir.join("test.zarr");
        adata.write::<Zarr, _>(file, None, None).unwrap();
    })
}

#[test]
fn test_save() {
    utils::test_save::<H5>();
    utils::test_save::<Zarr>();
}

#[test]
fn test_speacial_cases() {
    with_tmp_dir(|dir| {
        let file = dir.join("test.h5");
        let adata_gen = || AnnData::<H5>::new(&file).unwrap();
        utils::test_speacial_cases(|| adata_gen());

        let file = dir.join("test.zarr");
        let adata_gen = || AnnData::<Zarr>::new(&file).unwrap();
        utils::test_speacial_cases(|| adata_gen());

    })
}

#[test]
fn test_noncanonical() {
    with_tmp_dir(|dir| {
        let file = dir.join("test.h5");
        let adata_gen = || AnnData::<H5>::new(&file).unwrap();
        utils::test_noncanonical(|| adata_gen());

        let file = dir.join("test.zarr");
        let adata_gen = || AnnData::<Zarr>::new(&file).unwrap();
        utils::test_noncanonical(|| adata_gen());
    })
}

#[test]
fn test_io() {
    with_tmp_dir(|dir| {
        let file = dir.join("test.h5");
        let adata_gen = || AnnData::<H5>::new(&file).unwrap();
        utils::test_io(|| adata_gen());

        //let file = dir.join("test.zarr");
        //let adata_gen = || AnnData::<Zarr>::new(&file).unwrap();
        //utils::test_io(|| adata_gen());
    })
}

#[test]
fn test_index() {
    with_tmp_dir(|dir| {
        let file = dir.join("test.h5");
        let adata_gen = || AnnData::<H5>::new(&file).unwrap();
        utils::test_index(|| adata_gen());

        //let file = dir.join("test.zarr");
        //let adata_gen = || AnnData::<Zarr>::new(&file).unwrap();
        //utils::test_index(|| adata_gen());
    })
}

#[test]
fn test_iterator() {
    with_tmp_dir(|dir| {
        let file = dir.join("test.h5");
        let adata_gen = || AnnData::<H5>::new(&file).unwrap();
        utils::test_iterator(|| adata_gen());

        let file = dir.join("test.zarr");
        let adata_gen = || AnnData::<Zarr>::new(&file).unwrap();
        utils::test_iterator(|| adata_gen());
    })
}