Función que obtiene la unidad imaginaria de una potencia imaginaria

index.js

function getUnidadImaginaria(potencia) {
    let potenciasImaginarias = {
        0: '1',
        1: 'i',
        2: '-1',
        3: '-i',
        4: '1'
    };
    let restante = 0;

    while ((potencia % 4) != 0) {
        potencia--;
        restante++;
    }

    return potenciasImaginarias[restante];
}

// getUnidadImaginaria(100); (1)
// getUnidadImaginaria(501); (i)
// getUnidadImaginaria(123); (-i)