rustytorch/crates/specialized/rtx-fea/tests/quadrilateral9_finite_element_tdd.rs

//! TDD Tests for Quadrilateral9 FiniteElement Implementation
//! Following strict Red-Green-Refactor cycle
//! No mocks, stubs, or TODOs - only full implementations

#[cfg(test)]
mod quadrilateral9_tests {
    use nalgebra::Vector3;
    use rtx_fea::elements::shape_functions::shape_2d::Quadrilateral9;
    use rtx_fea::elements::{FiniteElement, NaturalCoords};
    use rtx_fea::mesh::ElementType;

    #[test]
    fn test_quad9_constructor() {
        // RED: Test that Quadrilateral9 can be constructed
        let quad9 = Quadrilateral9::new();

        // GREEN: Constructor should create valid instance
        assert_eq!(quad9.num_nodes(), 9);
    }

    #[test]
    fn test_quad9_element_type() {
        // RED: Test that Quadrilateral9 implements FiniteElement and returns correct type
        let quad9 = Quadrilateral9::new();

        // GREEN: Quadrilateral9 should return ElementType::Quad9
        assert_eq!(quad9.element_type(), ElementType::Quad9);
    }

    #[test]
    fn test_quad9_num_nodes() {
        // RED: Test that Quadrilateral9 correctly reports number of nodes
        let quad9 = Quadrilateral9::new();

        // GREEN: Quadrilateral9 has 9 nodes (4 corners + 4 mid-edges + 1 center)
        assert_eq!(quad9.num_nodes(), 9);
    }

    #[test]
    fn test_quad9_dimensions() {
        // RED: Test spatial and parametric dimensions
        let quad9 = Quadrilateral9::new();

        // GREEN: Quadrilateral9 is 2D element with 2D parametric space
        assert_eq!(quad9.spatial_dimension(), 2);
        assert_eq!(quad9.parametric_dimension(), 2);
    }

    #[test]
    fn test_quad9_shape_functions_at_corner() {
        // RED: Test shape function evaluation at corner node
        let quad9 = Quadrilateral9::new();

        // Test at corner (-1,-1) - should be 1 at node 0, 0 at others
        let coords = NaturalCoords::new_2d(-1.0, -1.0);
        let shape = quad9.shape_functions(&coords).unwrap();

        // GREEN: Verify shape function properties
        assert!((shape.value(0).unwrap() - 1.0).abs() < 1e-10);
        for i in 1..9 {
            assert!(shape.value(i).unwrap().abs() < 1e-10);
        }
    }

    #[test]
    fn test_quad9_partition_of_unity() {
        // RED: Test that shape functions sum to 1 everywhere
        let quad9 = Quadrilateral9::new();

        let test_points = vec![
            (0.0, 0.0),  // Center
            (0.5, 0.5),  // Random point
            (-0.5, 0.3), // Another random point
        ];

        for (x, y) in test_points {
            let coords = NaturalCoords::new_2d(x, y);
            let shape = quad9.shape_functions(&coords).unwrap();

            // GREEN: Sum of all shape functions should be 1
            let sum: f64 = (0..9).map(|i| shape.value(i).unwrap()).sum();
            assert!((sum - 1.0).abs() < 1e-10);
        }
    }

    #[test]
    fn test_quad9_jacobian() {
        // RED: Test Jacobian computation
        let quad9 = Quadrilateral9::new();

        // Create a regular quad with side length 2
        let node_coords = vec![
            Vector3::new(-1.0, -1.0, 0.0), // Corner nodes
            Vector3::new(1.0, -1.0, 0.0),
            Vector3::new(1.0, 1.0, 0.0),
            Vector3::new(-1.0, 1.0, 0.0),
            Vector3::new(0.0, -1.0, 0.0), // Mid-edge nodes
            Vector3::new(1.0, 0.0, 0.0),
            Vector3::new(0.0, 1.0, 0.0),
            Vector3::new(-1.0, 0.0, 0.0),
            Vector3::new(0.0, 0.0, 0.0), // Center node
        ];

        let coords = NaturalCoords::new_2d(0.0, 0.0);
        let jac = quad9.jacobian(&coords, &node_coords).unwrap();

        // GREEN: For regular quad, Jacobian at center should be diagonal
        assert!(jac.determinant.abs() > 1e-10); // Non-singular
        assert!((jac.determinant - 1.0).abs() < 1e-10); // Unit determinant for unit square
    }

    #[test]
    fn test_quad9_quadrature() {
        // RED: Test quadrature rule generation
        let quad9 = Quadrilateral9::new();

        // GREEN: Get default quadrature rule
        let quad_rule = quad9.quadrature_rule(None).unwrap();
        assert!(quad_rule.points.len() > 0);

        // For order 3, should have at least 9 points (3x3)
        let quad_rule_3 = quad9.quadrature_rule(Some(3)).unwrap();
        assert_eq!(quad_rule_3.points.len(), 9);
    }
}