Function test_multiple_groups

src/memory/processing/diffexp/mod.rs:1279–1313

()

    #[test]
    fn test_multiple_groups() -> anyhow::Result<()> {
        let adata = create_test_anndata_three_groups()?;

        rank_gene_groups(
            &adata,
            "group",
            None,
            Some(&["A", "B", "C"]),
            Some("multi_group"),
            Some(TestMethod::TTest(TTestType::Student)),
            Some(3),
            CorrectionMethod::BejaminiHochberg,
            Some(true),
            Some(1.0),
        )?;

        let uns = adata.uns();
        let scores = uns.get_data("rank_genes_groups_multi_group_scores")?;
        let _gene_names = uns.get_data("rank_genes_groups_multi_group_names")?;

        match scores.get_data()? {
            Data::ArrayData(ArrayData::DataFrame(df)) => {
                assert_eq!(df.width(), 3, "Should have results for 3 groups");
                assert_eq!(df.height(), 3, "Should have 3 genes per group");

                let column_names = df.get_column_names();
                assert!(column_names.iter().any(|name| name.as_str() == "A"));
                assert!(column_names.iter().any(|name| name.as_str() == "B"));
                assert!(column_names.iter().any(|name| name.as_str() == "C"));
            }
            _ => panic!("Expected DataFrame for scores"),
        }

        Ok(())
    }

    #[test]
    fn test_edge_cases() -> anyhow::Result<()> {