algeon-cudf 0.1.2

Safe libcudf execution component for Algeon
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

pub use crate::internal::testing_hooks::reassemble_dataframe_from_table_and_names_for_test;
use crate::{
    Error,
    core::{Result, schema::column_name_count_mismatch_error},
    frame::dataframe::{self, DataFrame, GpuRecordBatch},
};

/// Builds the canonical `DataFrame::new` column-count mismatch error shape for
/// integration-test helpers.
pub fn new_column_length_mismatch_error_for_test(column_name_count: usize, table_column_count: usize) -> Error {
    column_name_count_mismatch_error(
        "dataframe::new",
        "number of names must equal number of columns",
        column_name_count,
        table_column_count,
    )
}

/// Trigger the high-level DataFrame dropna subset-count limit validation
/// contract.
pub fn trigger_dataframe_dropna_subset_count_limit_error_for_test() -> Result<()> {
    dataframe::trigger_dataframe_dropna_subset_count_limit_error_for_test()
}

/// Trigger the DataFrame row-range start limit path used by integration tests.
pub fn trigger_dataframe_row_range_start_limit_error_for_test() -> Result<DataFrame> {
    dataframe::trigger_dataframe_row_range_start_limit_error_for_test()
}

/// Trigger the DataFrame row-range end limit path used by integration tests.
pub fn trigger_dataframe_row_range_end_limit_error_for_test() -> Result<DataFrame> {
    dataframe::trigger_dataframe_row_range_end_limit_error_for_test()
}

/// Trigger the DataFrame row-range overflow path used by integration tests.
pub fn trigger_dataframe_row_range_overflow_error_for_test() -> Result<DataFrame> {
    dataframe::trigger_dataframe_row_range_overflow_error_for_test()
}

/// Trigger the concat-by-name missing planned dtype invariant path used by
/// integration tests.
pub fn trigger_concat_by_name_missing_planned_dtype_for_test() -> Result<()> {
    dataframe::trigger_concat_by_name_missing_planned_dtype_for_test()
}

/// Reset the thread-local count of successful FixedSizeList structural proof
/// builds performed by `GpuRecordBatch::new`.
pub fn reset_fixed_size_list_structural_proof_build_count_for_test() {
    dataframe::reset_fixed_size_list_structural_proof_build_count_for_test();
}

/// Return the thread-local number of successful FixedSizeList structural
/// proof builds since the last reset.
pub fn fixed_size_list_structural_proof_build_count_for_test() -> usize {
    dataframe::fixed_size_list_structural_proof_build_count_for_test()
}

/// Attempt to combine one batch's retained FixedSizeList proof with another
/// batch's producer column. The production gate must reject the mismatch.
pub fn trigger_fixed_size_list_proof_identity_mismatch_for_test(
    producer: &GpuRecordBatch,
    proof_owner: &GpuRecordBatch,
    field_index: usize,
) -> Result<()> {
    producer.trigger_fixed_size_list_proof_identity_mismatch_for_test(proof_owner, field_index)
}

/// Attempt to combine one batch's uniform-list proof with another batch's
/// producer column. The production gate must reject the mismatch.
pub fn trigger_uniform_list_proof_identity_mismatch_for_test(
    producer: &GpuRecordBatch,
    proof_owner: &GpuRecordBatch,
    field_index: usize,
) -> Result<()> {
    producer.trigger_uniform_list_proof_identity_mismatch_for_test(proof_owner, field_index)
}