use cubecl_core as cubecl;
use cubecl_core::prelude::*;

#[cube]
pub fn literal<F: Float>(lhs: F) {
    let _ = lhs + F::from_int(5);
}

#[cube]
pub fn literal_float_no_decimals<F: Float>(lhs: F) {
    let _ = lhs + F::new(5.);
}

mod tests {
    use super::*;
    use cubecl_core::{
        cpa,
        ir::{Item, Variable},
    };

    type ElemType = f32;

    #[test]
    fn cube_literal_test() {
        let mut context = CubeContext::default();

        let lhs = context.create_local_binding(Item::new(ElemType::as_elem()));

        literal::expand::<ElemType>(&mut context, lhs.into());
        let scope = context.into_scope();

        assert_eq!(format!("{:?}", scope.operations), inline_macro_ref());
    }

    #[test]
    fn cube_literal_float_no_decimal_test() {
        let mut context = CubeContext::default();

        let lhs = context.create_local_binding(Item::new(ElemType::as_elem()));

        literal_float_no_decimals::expand::<ElemType>(&mut context, lhs.into());
        let scope = context.into_scope();

        assert_eq!(format!("{:?}", scope.operations), inline_macro_ref());
    }

    fn inline_macro_ref() -> String {
        let mut context = CubeContext::default();
        let item = Item::new(ElemType::as_elem());
        let lhs = context.create_local_binding(item);

        let mut scope = context.into_scope();
        let lhs: Variable = lhs.into();
        cpa!(scope, lhs = lhs + 5.0f32);

        format!("{:?}", scope.operations)
    }
}