JavaScript/Cache/test/Memoize.test.js at 8bce33d2659e101c79fd33e70caca66db6770dc0 · flexycode/JavaScript · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
import { memoize } from '../Memoize'

const fibonacci = (n) => {
  if (n < 2) {
    return n
  }

  return fibonacci(n - 2) + fibonacci(n - 1)
}

const factorial = (n) => {
  if (n === 0) {
    return 1
  }

  return n * factorial(n - 1)
}

describe('Memoize', () => {
  it('expects the fibonacci function to use the cache on the second call', () => {
    const memoFibonacci = memoize(fibonacci)

    expect(memoFibonacci(5)).toEqual(fibonacci(5))
    expect(memoFibonacci(5)).toEqual(5)
    expect(memoFibonacci(10)).toEqual(fibonacci(10))
    expect(memoFibonacci(10)).toEqual(55)
  })

  it('expects the factorial function to use the cache on the second call', () => {
    const memoFactorial = memoize(factorial)

    expect(memoFactorial(5)).toEqual(factorial(5))
    expect(memoFactorial(5)).toEqual(120)
    expect(memoFactorial(10)).toEqual(factorial(10))
    expect(memoFactorial(10)).toEqual(3628800)
  })
})