kermit-bench 0.1.0

Benchmarking utilities and benchmarks
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

use {crate::benchmark::BenchmarkConfig, clap::ValueEnum, std::str::FromStr};

pub mod oxford;

/// The set of available benchmarks.
///
/// Each variant maps to a [`BenchmarkConfig`] implementation that knows how to
/// download, transform, and validate its dataset.
#[derive(Copy, Clone, PartialEq, Eq, Debug, ValueEnum)]
pub enum Benchmark {
    Oxford,
}

impl Benchmark {
    pub fn from_name(name: &str) -> Result<Self, Box<dyn std::error::Error>> {
        if name == Benchmark::Oxford.name() {
            Ok(Self::Oxford)
        } else {
            Err(format!("Benchmark '{}' not found", name).into())
        }
    }

    pub fn names() -> Vec<String> { vec![Self::Oxford.name()] }

    pub fn name(self) -> String { self.config().metadata().name.to_string() }

    pub fn config(self) -> Box<dyn BenchmarkConfig + 'static> {
        match self {
            | Self::Oxford => Box::new(oxford::OxfordBenchmark),
        }
    }
}

impl FromStr for Benchmark {
    type Err = String;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        Self::from_name(s).map_err(|e| e.to_string())
    }
}