tensorlogic-adapters 0.1.0

//! Integration utilities for tensorlogic-compiler.
//!
//! This module provides utilities for exporting SymbolTable data to
//! tensorlogic-compiler's CompilerContext and for bidirectional synchronization.

use anyhow::Result;
use std::collections::HashMap;

use crate::{DomainHierarchy, DomainInfo, PredicateConstraints, PredicateInfo, SymbolTable};

/// Export utilities for compiler integration.
pub struct CompilerExport;

impl CompilerExport {
    /// Export domain information as a simple map for compiler consumption.
    ///
    /// This returns a HashMap that maps domain names to their cardinalities,
    /// suitable for direct use in compiler contexts.
    ///
    /// # Example
    ///
    /// ```rust
    /// use tensorlogic_adapters::{SymbolTable, DomainInfo, CompilerExport};
    ///
    /// let mut table = SymbolTable::new();
    /// table.add_domain(DomainInfo::new("Person", 100)).expect("unwrap");
    /// table.add_domain(DomainInfo::new("Location", 50)).expect("unwrap");
    ///
    /// let domain_map = CompilerExport::export_domains(&table);
    /// assert_eq!(domain_map.get("Person"), Some(&100));
    /// assert_eq!(domain_map.get("Location"), Some(&50));
    /// ```
    pub fn export_domains(table: &SymbolTable) -> HashMap<String, usize> {
        table
            .domains
            .iter()
            .map(|(name, info)| (name.clone(), info.cardinality))
            .collect()
    }

    /// Export predicate signatures for type checking.
    ///
    /// Returns a HashMap mapping predicate names to their argument domain lists.
    ///
    /// # Example
    ///
    /// ```rust
    /// use tensorlogic_adapters::{SymbolTable, DomainInfo, PredicateInfo, CompilerExport};
    ///
    /// let mut table = SymbolTable::new();
    /// table.add_domain(DomainInfo::new("Person", 100)).expect("unwrap");
    /// table.add_predicate(PredicateInfo::new(
    ///     "knows",
    ///     vec!["Person".to_string(), "Person".to_string()]
    /// )).expect("unwrap");
    ///
    /// let signatures = CompilerExport::export_predicate_signatures(&table);
    /// assert_eq!(signatures.get("knows"), Some(&vec!["Person".to_string(), "Person".to_string()]));
    /// ```
    pub fn export_predicate_signatures(table: &SymbolTable) -> HashMap<String, Vec<String>> {
        table
            .predicates
            .iter()
            .map(|(name, info)| (name.clone(), info.arg_domains.clone()))
            .collect()
    }

    /// Export variable bindings for scope analysis.
    ///
    /// Returns a HashMap mapping variable names to their domain types.
    ///
    /// # Example
    ///
    /// ```rust
    /// use tensorlogic_adapters::{SymbolTable, DomainInfo, CompilerExport};
    ///
    /// let mut table = SymbolTable::new();
    /// table.add_domain(DomainInfo::new("Person", 100)).expect("unwrap");
    /// table.bind_variable("x", "Person").expect("unwrap");
    /// table.bind_variable("y", "Person").expect("unwrap");
    ///
    /// let bindings = CompilerExport::export_variable_bindings(&table);
    /// assert_eq!(bindings.get("x"), Some(&"Person".to_string()));
    /// assert_eq!(bindings.get("y"), Some(&"Person".to_string()));
    /// ```
    pub fn export_variable_bindings(table: &SymbolTable) -> HashMap<String, String> {
        table
            .variables
            .iter()
            .map(|(var, domain)| (var.clone(), domain.clone()))
            .collect()
    }

    /// Create a complete export bundle for compiler initialization.
    ///
    /// Returns all three maps (domains, signatures, bindings) in a single structure.
    pub fn export_all(table: &SymbolTable) -> CompilerExportBundle {
        CompilerExportBundle {
            domains: Self::export_domains(table),
            predicate_signatures: Self::export_predicate_signatures(table),
            variable_bindings: Self::export_variable_bindings(table),
        }
    }
}

/// Complete export bundle for compiler integration.
///
/// This structure contains all information needed to initialize a compiler context
/// from a symbol table.
#[derive(Clone, Debug)]
pub struct CompilerExportBundle {
    /// Domain names mapped to cardinalities.
    pub domains: HashMap<String, usize>,
    /// Predicate names mapped to argument domain lists.
    pub predicate_signatures: HashMap<String, Vec<String>>,
    /// Variable names mapped to domain types.
    pub variable_bindings: HashMap<String, String>,
}

impl CompilerExportBundle {
    /// Create an empty export bundle.
    pub fn new() -> Self {
        Self {
            domains: HashMap::new(),
            predicate_signatures: HashMap::new(),
            variable_bindings: HashMap::new(),
        }
    }

    /// Check if the bundle is empty.
    pub fn is_empty(&self) -> bool {
        self.domains.is_empty()
            && self.predicate_signatures.is_empty()
            && self.variable_bindings.is_empty()
    }
}

impl Default for CompilerExportBundle {
    fn default() -> Self {
        Self::new()
    }
}

/// Import utilities for reverse synchronization.
pub struct CompilerImport;

impl CompilerImport {
    /// Import domain information from a compiler context back into a symbol table.
    ///
    /// This is useful for synchronizing state after compilation.
    ///
    /// # Example
    ///
    /// ```rust
    /// use tensorlogic_adapters::{SymbolTable, CompilerImport};
    /// use std::collections::HashMap;
    ///
    /// let mut domains = HashMap::new();
    /// domains.insert("Person".to_string(), 100);
    /// domains.insert("Location".to_string(), 50);
    ///
    /// let mut table = SymbolTable::new();
    /// CompilerImport::import_domains(&mut table, &domains).expect("unwrap");
    ///
    /// assert!(table.get_domain("Person").is_some());
    /// assert!(table.get_domain("Location").is_some());
    /// ```
    pub fn import_domains(table: &mut SymbolTable, domains: &HashMap<String, usize>) -> Result<()> {
        for (name, cardinality) in domains {
            table.add_domain(DomainInfo::new(name.clone(), *cardinality))?;
        }
        Ok(())
    }

    /// Import predicate signatures from a compiler context.
    ///
    /// # Example
    ///
    /// ```rust
    /// use tensorlogic_adapters::{SymbolTable, DomainInfo, CompilerImport};
    /// use std::collections::HashMap;
    ///
    /// let mut table = SymbolTable::new();
    /// table.add_domain(DomainInfo::new("Person", 100)).expect("unwrap");
    ///
    /// let mut signatures = HashMap::new();
    /// signatures.insert("knows".to_string(), vec!["Person".to_string(), "Person".to_string()]);
    ///
    /// CompilerImport::import_predicates(&mut table, &signatures).expect("unwrap");
    ///
    /// assert!(table.get_predicate("knows").is_some());
    /// ```
    pub fn import_predicates(
        table: &mut SymbolTable,
        signatures: &HashMap<String, Vec<String>>,
    ) -> Result<()> {
        for (name, arg_domains) in signatures {
            table.add_predicate(PredicateInfo::new(name.clone(), arg_domains.clone()))?;
        }
        Ok(())
    }

    /// Import variable bindings from a compiler context.
    ///
    /// # Example
    ///
    /// ```rust
    /// use tensorlogic_adapters::{SymbolTable, DomainInfo, CompilerImport};
    /// use std::collections::HashMap;
    ///
    /// let mut table = SymbolTable::new();
    /// table.add_domain(DomainInfo::new("Person", 100)).expect("unwrap");
    ///
    /// let mut bindings = HashMap::new();
    /// bindings.insert("x".to_string(), "Person".to_string());
    /// bindings.insert("y".to_string(), "Person".to_string());
    ///
    /// CompilerImport::import_variables(&mut table, &bindings).expect("unwrap");
    ///
    /// assert_eq!(table.get_variable_domain("x"), Some("Person"));
    /// assert_eq!(table.get_variable_domain("y"), Some("Person"));
    /// ```
    pub fn import_variables(
        table: &mut SymbolTable,
        bindings: &HashMap<String, String>,
    ) -> Result<()> {
        for (var, domain) in bindings {
            table.bind_variable(var, domain)?;
        }
        Ok(())
    }

    /// Import a complete bundle into a symbol table.
    pub fn import_all(table: &mut SymbolTable, bundle: &CompilerExportBundle) -> Result<()> {
        Self::import_domains(table, &bundle.domains)?;
        Self::import_predicates(table, &bundle.predicate_signatures)?;
        Self::import_variables(table, &bundle.variable_bindings)?;
        Ok(())
    }
}

/// Bidirectional synchronization utilities.
pub struct SymbolTableSync;

impl SymbolTableSync {
    /// Synchronize a symbol table with compiler data, merging information.
    ///
    /// This performs a two-way sync:
    /// 1. Exports current symbol table state
    /// 2. Imports compiler context data
    /// 3. Returns the merged export bundle
    pub fn sync_with_compiler(
        table: &mut SymbolTable,
        compiler_bundle: &CompilerExportBundle,
    ) -> Result<CompilerExportBundle> {
        // First, import compiler data into the table
        CompilerImport::import_all(table, compiler_bundle)?;

        // Then, export the merged state
        Ok(CompilerExport::export_all(table))
    }

    /// Validate that a compiler bundle is compatible with a symbol table.
    ///
    /// Checks that all referenced domains exist.
    pub fn validate_bundle(
        table: &SymbolTable,
        bundle: &CompilerExportBundle,
    ) -> Result<ValidationResult> {
        let mut errors = Vec::new();
        let mut warnings = Vec::new();

        // Check predicate signatures reference existing domains
        for (pred_name, arg_domains) in &bundle.predicate_signatures {
            for domain in arg_domains {
                if !table.domains.contains_key(domain) && !bundle.domains.contains_key(domain) {
                    errors.push(format!(
                        "Predicate '{}' references unknown domain '{}'",
                        pred_name, domain
                    ));
                }
            }
        }

        // Check variable bindings reference existing domains
        for (var_name, domain) in &bundle.variable_bindings {
            if !table.domains.contains_key(domain) && !bundle.domains.contains_key(domain) {
                errors.push(format!(
                    "Variable '{}' references unknown domain '{}'",
                    var_name, domain
                ));
            }
        }

        // Warn about unused domains
        for domain_name in bundle.domains.keys() {
            let used_in_predicates = bundle
                .predicate_signatures
                .values()
                .any(|args| args.contains(domain_name));
            let used_in_variables = bundle.variable_bindings.values().any(|d| d == domain_name);

            if !used_in_predicates && !used_in_variables {
                warnings.push(format!(
                    "Domain '{}' is defined but never used",
                    domain_name
                ));
            }
        }

        Ok(ValidationResult { errors, warnings })
    }
}

/// Result of bundle validation.
#[derive(Clone, Debug)]
pub struct ValidationResult {
    /// Validation errors (must be empty for valid bundles).
    pub errors: Vec<String>,
    /// Validation warnings (non-critical issues).
    pub warnings: Vec<String>,
}

impl ValidationResult {
    /// Check if validation passed (no errors).
    pub fn is_valid(&self) -> bool {
        self.errors.is_empty()
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_export_domains() {
        let mut table = SymbolTable::new();
        table
            .add_domain(DomainInfo::new("Person", 100))
            .expect("unwrap");
        table
            .add_domain(DomainInfo::new("Location", 50))
            .expect("unwrap");

        let domains = CompilerExport::export_domains(&table);
        assert_eq!(domains.get("Person"), Some(&100));
        assert_eq!(domains.get("Location"), Some(&50));
    }

    #[test]
    fn test_export_predicate_signatures() {
        let mut table = SymbolTable::new();
        table
            .add_domain(DomainInfo::new("Person", 100))
            .expect("unwrap");
        table
            .add_predicate(PredicateInfo::new(
                "knows",
                vec!["Person".to_string(), "Person".to_string()],
            ))
            .expect("unwrap");

        let signatures = CompilerExport::export_predicate_signatures(&table);
        assert_eq!(
            signatures.get("knows"),
            Some(&vec!["Person".to_string(), "Person".to_string()])
        );
    }

    #[test]
    fn test_export_all() {
        let mut table = SymbolTable::new();
        table
            .add_domain(DomainInfo::new("Person", 100))
            .expect("unwrap");
        table
            .add_predicate(PredicateInfo::new("knows", vec!["Person".to_string()]))
            .expect("unwrap");
        table.bind_variable("x", "Person").expect("unwrap");

        let bundle = CompilerExport::export_all(&table);
        assert_eq!(bundle.domains.len(), 1);
        assert_eq!(bundle.predicate_signatures.len(), 1);
        assert_eq!(bundle.variable_bindings.len(), 1);
    }

    #[test]
    fn test_import_domains() {
        let mut domains = HashMap::new();
        domains.insert("Person".to_string(), 100);

        let mut table = SymbolTable::new();
        CompilerImport::import_domains(&mut table, &domains).expect("unwrap");

        assert!(table.get_domain("Person").is_some());
    }

    #[test]
    fn test_import_predicates() {
        let mut table = SymbolTable::new();
        table
            .add_domain(DomainInfo::new("Person", 100))
            .expect("unwrap");

        let mut signatures = HashMap::new();
        signatures.insert("knows".to_string(), vec!["Person".to_string()]);

        CompilerImport::import_predicates(&mut table, &signatures).expect("unwrap");
        assert!(table.get_predicate("knows").is_some());
    }

    #[test]
    fn test_validation_invalid_domain_reference() {
        let table = SymbolTable::new();
        let mut bundle = CompilerExportBundle::new();
        bundle
            .predicate_signatures
            .insert("knows".to_string(), vec!["UnknownDomain".to_string()]);

        let result = SymbolTableSync::validate_bundle(&table, &bundle).expect("unwrap");
        assert!(!result.is_valid());
        assert!(!result.errors.is_empty());
    }

    #[test]
    fn test_validation_unused_domain_warning() {
        let table = SymbolTable::new();
        let mut bundle = CompilerExportBundle::new();
        bundle.domains.insert("UnusedDomain".to_string(), 100);

        let result = SymbolTableSync::validate_bundle(&table, &bundle).expect("unwrap");
        assert!(result.is_valid()); // Still valid, just has warnings
        assert!(!result.warnings.is_empty());
    }

    #[test]
    fn test_sync_with_compiler() {
        let mut table = SymbolTable::new();
        table
            .add_domain(DomainInfo::new("Person", 100))
            .expect("unwrap");

        let mut bundle = CompilerExportBundle::new();
        bundle.domains.insert("Location".to_string(), 50);

        let result = SymbolTableSync::sync_with_compiler(&mut table, &bundle).expect("unwrap");

        // Table should now have both domains
        assert!(table.get_domain("Person").is_some());
        assert!(table.get_domain("Location").is_some());

        // Result should include both
        assert_eq!(result.domains.len(), 2);
    }
}

/// Advanced export utilities for compiler integration with type systems.
///
/// This extends the basic CompilerExport with support for advanced type system
/// features including refinement types, dependent types, linear types, and effects.
pub struct CompilerExportAdvanced;

impl CompilerExportAdvanced {
    /// Export domain hierarchy information for subtype checking.
    ///
    /// Returns the complete subtype relationships from the hierarchy.
    ///
    /// # Example
    ///
    /// ```rust
    /// use tensorlogic_adapters::{DomainHierarchy, CompilerExportAdvanced};
    ///
    /// let mut hierarchy = DomainHierarchy::new();
    /// hierarchy.add_subtype("Student", "Person");
    /// hierarchy.add_subtype("Person", "Agent");
    ///
    /// let relationships = CompilerExportAdvanced::export_hierarchy(&hierarchy);
    /// assert!(relationships.contains_key("Student"));
    /// ```
    pub fn export_hierarchy(hierarchy: &DomainHierarchy) -> HashMap<String, Vec<String>> {
        let mut result = HashMap::new();

        for domain in hierarchy.all_domains() {
            // Get all ancestors (transitive closure of parent relationships)
            let ancestors = hierarchy.get_ancestors(&domain);
            if !ancestors.is_empty() {
                result.insert(domain, ancestors);
            }
        }

        result
    }

    /// Export predicate constraints for validation and optimization.
    ///
    /// Returns a map of predicate names to their constraint specifications.
    ///
    /// # Example
    ///
    /// ```rust
    /// use tensorlogic_adapters::{SymbolTable, DomainInfo, PredicateInfo, PredicateConstraints, CompilerExportAdvanced};
    ///
    /// let mut table = SymbolTable::new();
    /// table.add_domain(DomainInfo::new("Person", 100)).expect("unwrap");
    ///
    /// let mut pred = PredicateInfo::new("age", vec!["Person".to_string()]);
    /// pred.constraints = Some(PredicateConstraints::new());
    /// table.add_predicate(pred).expect("unwrap");
    ///
    /// let constraints = CompilerExportAdvanced::export_constraints(&table);
    /// assert!(constraints.contains_key("age"));
    /// ```
    pub fn export_constraints(table: &SymbolTable) -> HashMap<String, PredicateConstraints> {
        table
            .predicates
            .iter()
            .filter_map(|(name, info)| info.constraints.as_ref().map(|c| (name.clone(), c.clone())))
            .collect()
    }

    /// Export refinement types for compile-time validation.
    ///
    /// Returns refinement type specifications that can be used for
    /// static analysis and runtime validation.
    pub fn export_refinement_types() -> HashMap<String, String> {
        // Export built-in refinement types
        // In a real implementation, this would be populated from a RefinementRegistry
        let mut types = HashMap::new();
        types.insert("PositiveInt".to_string(), "{ x: Int | x > 0 }".to_string());
        types.insert(
            "Probability".to_string(),
            "{ x: Float | 0.0 <= x <= 1.0 }".to_string(),
        );
        types.insert(
            "NonZeroFloat".to_string(),
            "{ x: Float | x != 0.0 }".to_string(),
        );
        types
    }

    /// Export dependent type information for dimension tracking.
    ///
    /// Returns dependent type specifications for compile-time dimension checking.
    pub fn export_dependent_types() -> HashMap<String, String> {
        // Export common dependent type patterns
        let mut types = HashMap::new();
        types.insert("Vector".to_string(), "Vector<T, n>".to_string());
        types.insert("Matrix".to_string(), "Matrix<T, m, n>".to_string());
        types.insert("Tensor3D".to_string(), "Tensor<T, i, j, k>".to_string());
        types.insert("SquareMatrix".to_string(), "Matrix<T, n, n>".to_string());
        types
    }

    /// Export linear type resources for lifetime tracking.
    ///
    /// Returns linear type specifications for resource management.
    pub fn export_linear_types() -> HashMap<String, String> {
        // Export linear type specifications
        let mut types = HashMap::new();
        types.insert("GpuTensor".to_string(), "Linear".to_string());
        types.insert("FileHandle".to_string(), "Linear".to_string());
        types.insert("NetworkSocket".to_string(), "Linear".to_string());
        types.insert("MutableReference".to_string(), "Affine".to_string());
        types.insert("LogMessage".to_string(), "Relevant".to_string());
        types
    }

    /// Export effect information for effect tracking.
    ///
    /// Returns effect specifications for compile-time effect checking.
    pub fn export_effects() -> HashMap<String, Vec<String>> {
        // Export common effect patterns
        let mut effects = HashMap::new();
        effects.insert("read_file".to_string(), vec!["IO".to_string()]);
        effects.insert(
            "allocate_gpu".to_string(),
            vec!["GPU".to_string(), "State".to_string()],
        );
        effects.insert(
            "train_model".to_string(),
            vec!["State".to_string(), "NonDet".to_string()],
        );
        effects.insert(
            "fetch_data".to_string(),
            vec!["IO".to_string(), "Exception".to_string()],
        );
        effects
    }

    /// Create a complete advanced export bundle.
    ///
    /// Returns all advanced type system information in a single structure.
    pub fn export_all_advanced(
        table: &SymbolTable,
        hierarchy: Option<&DomainHierarchy>,
    ) -> AdvancedExportBundle {
        AdvancedExportBundle {
            hierarchy: hierarchy.map(Self::export_hierarchy),
            constraints: Self::export_constraints(table),
            refinement_types: Self::export_refinement_types(),
            dependent_types: Self::export_dependent_types(),
            linear_types: Self::export_linear_types(),
            effects: Self::export_effects(),
        }
    }
}

/// Advanced export bundle containing type system information.
#[derive(Clone, Debug)]
pub struct AdvancedExportBundle {
    /// Domain hierarchy (subtype relationships).
    pub hierarchy: Option<HashMap<String, Vec<String>>>,
    /// Predicate constraints.
    pub constraints: HashMap<String, PredicateConstraints>,
    /// Refinement type specifications.
    pub refinement_types: HashMap<String, String>,
    /// Dependent type specifications.
    pub dependent_types: HashMap<String, String>,
    /// Linear type specifications.
    pub linear_types: HashMap<String, String>,
    /// Effect specifications.
    pub effects: HashMap<String, Vec<String>>,
}

impl AdvancedExportBundle {
    /// Create an empty advanced export bundle.
    pub fn new() -> Self {
        Self {
            hierarchy: None,
            constraints: HashMap::new(),
            refinement_types: HashMap::new(),
            dependent_types: HashMap::new(),
            linear_types: HashMap::new(),
            effects: HashMap::new(),
        }
    }

    /// Check if the bundle is empty.
    pub fn is_empty(&self) -> bool {
        self.hierarchy.is_none()
            && self.constraints.is_empty()
            && self.refinement_types.is_empty()
            && self.dependent_types.is_empty()
            && self.linear_types.is_empty()
            && self.effects.is_empty()
    }
}

impl Default for AdvancedExportBundle {
    fn default() -> Self {
        Self::new()
    }
}

/// Complete compiler integration bundle combining basic and advanced exports.
#[derive(Clone, Debug)]
pub struct CompleteExportBundle {
    /// Basic export bundle (domains, predicates, variables).
    pub basic: CompilerExportBundle,
    /// Advanced type system export bundle.
    pub advanced: AdvancedExportBundle,
}

impl CompleteExportBundle {
    /// Create a new complete export bundle.
    pub fn new(basic: CompilerExportBundle, advanced: AdvancedExportBundle) -> Self {
        Self { basic, advanced }
    }

    /// Export everything from a symbol table and hierarchy.
    pub fn from_symbol_table(table: &SymbolTable, hierarchy: Option<&DomainHierarchy>) -> Self {
        Self {
            basic: CompilerExport::export_all(table),
            advanced: CompilerExportAdvanced::export_all_advanced(table, hierarchy),
        }
    }

    /// Check if the bundle is completely empty.
    pub fn is_empty(&self) -> bool {
        self.basic.is_empty() && self.advanced.is_empty()
    }
}

#[cfg(test)]
mod advanced_tests {
    use super::*;

    #[test]
    fn test_export_hierarchy() {
        let mut hierarchy = DomainHierarchy::new();
        hierarchy.add_subtype("Student", "Person");
        hierarchy.add_subtype("Person", "Agent");

        let relationships = CompilerExportAdvanced::export_hierarchy(&hierarchy);

        assert!(relationships.contains_key("Student"));
        assert!(relationships.contains_key("Person"));
    }

    #[test]
    fn test_export_constraints() {
        let mut table = SymbolTable::new();
        table
            .add_domain(DomainInfo::new("Person", 100))
            .expect("unwrap");

        let mut pred = PredicateInfo::new("age", vec!["Person".to_string()]);
        pred.constraints = Some(PredicateConstraints::new());
        table.add_predicate(pred).expect("unwrap");

        let constraints = CompilerExportAdvanced::export_constraints(&table);
        assert!(constraints.contains_key("age"));
    }

    #[test]
    fn test_export_refinement_types() {
        let types = CompilerExportAdvanced::export_refinement_types();
        assert!(types.contains_key("PositiveInt"));
        assert!(types.contains_key("Probability"));
    }

    #[test]
    fn test_export_dependent_types() {
        let types = CompilerExportAdvanced::export_dependent_types();
        assert!(types.contains_key("Vector"));
        assert!(types.contains_key("Matrix"));
    }

    #[test]
    fn test_export_linear_types() {
        let types = CompilerExportAdvanced::export_linear_types();
        assert!(types.contains_key("GpuTensor"));
        assert!(types.contains_key("FileHandle"));
    }

    #[test]
    fn test_export_effects() {
        let effects = CompilerExportAdvanced::export_effects();
        assert!(effects.contains_key("read_file"));
        assert!(effects.contains_key("allocate_gpu"));
    }

    #[test]
    fn test_export_all_advanced() {
        let mut table = SymbolTable::new();
        table
            .add_domain(DomainInfo::new("Person", 100))
            .expect("unwrap");

        let mut hierarchy = DomainHierarchy::new();
        hierarchy.add_subtype("Student", "Person");

        let bundle = CompilerExportAdvanced::export_all_advanced(&table, Some(&hierarchy));

        assert!(bundle.hierarchy.is_some());
        assert!(!bundle.refinement_types.is_empty());
        assert!(!bundle.dependent_types.is_empty());
    }

    #[test]
    fn test_complete_export_bundle() {
        let mut table = SymbolTable::new();
        table
            .add_domain(DomainInfo::new("Person", 100))
            .expect("unwrap");

        let mut hierarchy = DomainHierarchy::new();
        hierarchy.add_subtype("Student", "Person");

        let bundle = CompleteExportBundle::from_symbol_table(&table, Some(&hierarchy));

        assert!(!bundle.is_empty());
        assert_eq!(bundle.basic.domains.len(), 1);
        assert!(bundle.advanced.hierarchy.is_some());
    }

    #[test]
    fn test_advanced_bundle_empty() {
        let bundle = AdvancedExportBundle::new();
        assert!(bundle.is_empty());
    }
}