//! Quick smoke test for new architectures

#[cfg(test)]
mod quick_tests {
    use rtx_vision::architectures::*;
    use rtx_vision::{Device, Tensor};

    #[test]
    fn test_basic_functionality() {
        let device = Device::cpu();

        // Test ResNet
        let resnet = ResNet::new(ResNetConfig::resnet18()).unwrap();
        let input = Tensor::randn(&[1, 3, 224, 224], &device).unwrap();
        assert!(resnet.forward(&input).is_ok());

        // Test VGG
        let vgg = VGG::new(VGGConfig::vgg11()).unwrap();
        assert!(vgg.forward(&input).is_ok());

        // Test DenseNet (commented out - needs stub implementation)
        // let densenet = DenseNet::new(DenseNetConfig::densenet121()).unwrap();
        // assert!(densenet.forward(&input).is_ok());

        // Test MobileNet (commented out - needs stub implementation)
        // let mobilenet = MobileNet::new(MobileNetConfig::default()).unwrap();
        // assert!(mobilenet.forward(&input).is_ok());
    }
}