Следите за функциями модуля в шутку

Question

Я пишу тесты с помощью jest, в которых я хочу следить за некоторыми функциями lodash, которые я импортирую отдельно в модуль (в отличие от импорта всего модуля lodash как _), например,

/** matrix.js **/
import shuffle from 'lodash/shuffle'
import pick from 'lodash/pick'

// ...

/**
 * Shuffles the order of the rows in the matrix. If a column/variable name
 * is specified, only the rows in this column are shuffled.
 *
 * @export
 * @param {array} matrix The matrix to be shuffled
 * @param {array} columns  Array containing the variable/column to be shuffled
 * @returns {array}
 */
export function shuffleVert (matrix, columns) {
  if (typeof (columns) === 'undefined' || (isArray(columns) && columns.length === 0)) {
    return shuffle(matrix)
  } else if (!isArray(columns)) {
    throw new TypeError('Invalid argument for columns specified to shuffleVert. Expects an array containing column names')
  } else {
    let grouped = unstack(matrix)
    let cols = pick(grouped, columns)
    cols = Object.entries(cols).reduce((prev, [key, values]) => {
      prev[key] = shuffle(values)
      return prev
    }, {})
    return stack({ ...grouped, ...cols })
 }

Функция shuffleVert тасует все строки матрицы или только строки указанных столбцов.Поскольку сложно тестировать функции со случайным выводом, насколько мне известно, я просто хочу проверить, были ли вызваны функции перемешивания и выбора lodash внутри протестированной функции.

В настоящее время я реализовал работающего шпионапроцедура в моем тестовом модуле, но я не думаю, что она обычная или эффективная, и я просто думаю, что должен быть лучший способ сделать это ...

/* matrix.test.js */
import {
  shuffleVert,
} from 'matrix'

/** Generate a mock functions to spy on lodash */
const mockShuffle = jest.fn()
jest.mock('lodash/shuffle', () => a => {
  const shuffle = jest.requireActual('lodash/shuffle')
  mockShuffle()
  return shuffle(a)
})

const mockPick = jest.fn()
jest.mock('lodash/pick', () => (a, b) => {
  const pick = jest.requireActual('lodash/pick')
  mockPick()
  return pick(a, b)
})

describe('reverseRows', () => {
  let srcMatrix
  beforeEach(() => {
    srcMatrix = [
      { number: 1, word: 'one' },
      { number: 2, word: 'two' },
      { number: 3, word: 'three' }
    ]
    mockShuffle.mockClear()
    mockPick.mockClear()
  })

  it('should shuffle the rows of the entire matrix with no argument for columns', () => {
    shuffleVert(srcMatrix)
    // 2 is weird, but seems correct.
    // It appears the shuffle function calls itself recursively
    expect(mockShuffle).toHaveBeenCalledTimes(2)
  })

  it('should only shuffle the rows of columns that were specified', () => {
    shuffleVert(srcMatrix, ['word'])
    expect(mockShuffle).toHaveBeenCalledTimes(2)
    expect(mockPick).toHaveBeenCalledTimes(2)
  })
})

Я знаю, что шут имеет spyOnфункциональность, но это только работает на объектные методы, поэтому

import * as lodash from 'lodash'
const shuffleSpy = jest.spyOn(lodash, 'shuffle')

приводит к ошибке Cannot spyOn on a primitive value; undefined given

Как правило, лучший способ шпионить за методами модуля?Есть ли лучшая реализация для того, что я пытаюсь достичь?Или это уже путь?