FilesExpand file tree

 master

/

test_mini_example.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

54 lines (37 loc) · 1.09 KB

 master

/

test_mini_example.py

Top

File metadata and controls

• Code
• Blame

54 lines (37 loc) · 1.09 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

import numpy as np

import pytest

from ctc_decoder import beam_search

from ctc_decoder import best_path

from ctc_decoder import prefix_search

from ctc_decoder import probability, loss

from ctc_decoder import token_passing

@pytest.fixture

def mat():

return np.array([[0.4, 0, 0.6], [0.4, 0, 0.6]])

@pytest.fixture

def chars():

return 'ab'

def test_beam_search(mat, chars):

expected = 'a'

actual = beam_search(mat, chars)

assert actual == expected

def test_best_path(mat, chars):

expected = ''

actual = best_path(mat, chars)

assert actual == expected

def test_token_passing(mat, chars):

expected = 'a'

actual = token_passing(mat, chars, ['a', 'b', 'ab', 'ba'])

assert actual == expected

def test_prefix_search(mat, chars):

expected = 'a'

actual = prefix_search(mat, chars)

assert actual == expected

def test_probability(mat, chars):

expected = 0.64

actual = probability(mat, 'a', chars)

assert actual == expected

def test_loss(mat, chars):

expected = -np.log(0.64)

actual = loss(mat, 'a', chars)

assert actual == expected