dl-asr/utils/ops_test.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#
# Copyright 2016-2099 Ailemon.net
#
# This file is part of ASRT Speech Recognition Tool.
#
# ASRT is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
# ASRT is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with ASRT.  If not, see <https://www.gnu.org/licenses/>.
# ============================================================================

"""
@author: nl8590687
ops.py单元测试
"""

import pytest
from ops import get_edit_distance, ctc_decode_delete_tail_blank, ctc_decode_stream


class TestGetEditDistance:
    def test_1(self):
        examle_input = ["abc", "abc"]
        examle_output = 0
        result = get_edit_distance(examle_input[0], examle_input[1])
        assert result == examle_output

    def test_2(self):
        examle_input = ["abc", "adc"]
        examle_output = 1
        result = get_edit_distance(examle_input[0], examle_input[1])
        assert result == examle_output

    def test_3(self):
        examle_input = ["abc", "a"]
        examle_output = 2
        result = get_edit_distance(examle_input[0], examle_input[1])
        assert result == examle_output

    def test_4(self):
        examle_input = ["abc", "addce"]
        examle_output = 3
        result = get_edit_distance(examle_input[0], examle_input[1])
        assert result == examle_output

    def test_5(self):
        examle_input = ["abc", ""]
        examle_output = 3
        result = get_edit_distance(examle_input[0], examle_input[1])
        assert result == examle_output

    def test_6(self):
        examle_input = ["", ""]
        examle_output = 0
        result = get_edit_distance(examle_input[0], examle_input[1])
        assert result == examle_output


class TestCtcDecodeDeleteTailBlank:
    def test_1(self):
        examle_input = [1, 2, 3, 4, 5, -1, -1, -1, -1, -1, -1]
        examle_output = [1, 2, 3, 4, 5]
        result = ctc_decode_delete_tail_blank(examle_input)
        assert result == examle_output

    def test_2(self):
        examle_input = [1, 2, 3, 4, 5, -1]
        examle_output = [1, 2, 3, 4, 5]
        result = ctc_decode_delete_tail_blank(examle_input)
        assert result == examle_output

    def test_3(self):
        examle_input = [1, 2, 3, 4, 5]
        examle_output = [1, 2, 3, 4, 5]
        result = ctc_decode_delete_tail_blank(examle_input)
        assert result == examle_output

    def test_4(self):
        examle_input = [-1, -1, -1, -1]
        examle_output = []
        result = ctc_decode_delete_tail_blank(examle_input)
        assert result == examle_output

    def test_5(self):
        examle_input = [-1]
        examle_output = []
        result = ctc_decode_delete_tail_blank(examle_input)
        assert result == examle_output


class TestCtcDecodeStream:
    def test_1(self):
        examle_input = [-1, -1, -1, 1, 1, 1, -1, -1, 2, 2, 2, 3, 3, -1, -1, -1]
        examle_output = [1, 2, 3, -1]
        result = []
        res, tmp = ctc_decode_stream(examle_input)
        result.append(res)
        while len(tmp) > 0:
            res, tmp = ctc_decode_stream(tmp)
            result.append(res)
        assert result == examle_output

    def test_2(self):
        examle_input = [-1, -1, -1, -1, -1]
        examle_output = [-1]
        result = []
        res, tmp = ctc_decode_stream(examle_input)
        result.append(res)
        while len(tmp) > 0:
            res, tmp = ctc_decode_stream(tmp)
            result.append(res)
        assert result == examle_output