//! Simple ResNet test without dependencies

#[cfg(test)]
mod simple_tests {
    use rtx_vision::architectures::{ResNet, ResNetConfig, ResNetVariant};
    use rtx_vision::{Device, Tensor};

    #[test]
    fn test_resnet_basic_creation() {
        let device = Device::cpu();
        let config = ResNetConfig::resnet18();
        let result = ResNet::new(config);
        assert!(result.is_ok());
    }

    #[test]
    fn test_resnet_config_variants() {
        let variants = vec![
            ResNetConfig::resnet18(),
            ResNetConfig::resnet34(),
            ResNetConfig::resnet50(),
            ResNetConfig::resnet101(),
            ResNetConfig::resnet152(),
        ];

        for config in variants {
            let device = Device::cpu();
            let result = ResNet::new(config);
            assert!(result.is_ok());
        }
    }

    #[test]
    fn test_resnet_forward_basic() {
        let device = Device::cpu();
        let config = ResNetConfig::resnet18();
        let model = ResNet::new(config).unwrap();

        let input = Tensor::randn(&[1, 3, 224, 224], &device).unwrap();
        let result = model.forward(&input);

        assert!(result.is_ok());
    }
}