def METHOD_NAME():
    assert register_type_for_type(builtin.i32) == riscv.IntRegisterType
    assert (
        register_type_for_type(
            memref.MemRefType.from_element_type_and_shape(builtin.f32, [1, 2, 3])
        )
        == riscv.IntRegisterType
    )
    assert (
        register_type_for_type(
            memref.MemRefType.from_element_type_and_shape(builtin.i32, [1, 2, 3])
        )
        == riscv.IntRegisterType
    )
    assert register_type_for_type(builtin.f32) == riscv.FloatRegisterType
    assert register_type_for_type(builtin.f64) == riscv.FloatRegisterType