xtalforge-core 0.0.2

Core crystal-structure types for xtalforge: elements, lattices, sites and structures.
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

//! Integration tests for functional groups (Task 1.13).

#![allow(
    clippy::unwrap_used,
    clippy::expect_used,
    clippy::panic,
    clippy::indexing_slicing
)]

use xtalforge_core::{GroupKind, Lattice, Site, StructureBuilder, find_groups, group_counts};

fn build(sites: &[(&str, [f64; 3])]) -> xtalforge_core::Structure {
    let mut builder = StructureBuilder::new("groups", Lattice::cubic(10.0).unwrap());
    for (index, (element, frac)) in sites.iter().enumerate() {
        builder = builder.site(Site::new(format!("{element}{index}"), element, *frac).unwrap());
    }
    builder.build().unwrap()
}

#[test]
fn finds_hydroxyl_in_methanol() {
    let structure = build(&[
        ("C", [0.5, 0.5, 0.5]),
        ("O", [0.56, 0.5, 0.5]),
        ("H", [0.656, 0.5, 0.5]),
    ]);
    let groups = find_groups(&structure).unwrap();
    assert!(groups.iter().any(|group| group.kind == GroupKind::Hydroxyl));
    assert_eq!(
        groups
            .iter()
            .filter(|group| group.kind == GroupKind::Hydroxyl)
            .count(),
        1
    );
}

#[test]
fn finds_carboxylate_and_fluoride() {
    let carboxylate = build(&[
        ("C", [0.5, 0.5, 0.5]),
        ("O", [0.56, 0.5, 0.5]),
        ("O", [0.44, 0.5, 0.5]),
    ]);
    assert_eq!(
        find_groups(&carboxylate)
            .unwrap()
            .iter()
            .filter(|group| group.kind == GroupKind::Carboxylate)
            .count(),
        1
    );

    let fluoride = build(&[("C", [0.5, 0.5, 0.5]), ("F", [0.56, 0.5, 0.5])]);
    assert!(
        find_groups(&fluoride)
            .unwrap()
            .iter()
            .any(|group| group.kind == GroupKind::Fluoride)
    );
}

#[test]
fn counts_groups() {
    let structure = build(&[
        ("C", [0.5, 0.5, 0.5]),
        ("O", [0.56, 0.5, 0.5]),
        ("H", [0.656, 0.5, 0.5]),
    ]);
    let counts = group_counts(&structure).unwrap();
    assert!(counts.contains(&(GroupKind::Hydroxyl, 1)));
}