from basis import ScoreBasis


class NB_PESQ(ScoreBasis):
    def __init__(self):
        super(NB_PESQ, self).__init__(name='NB_PESQ')
        self.intrusive = False
        self.score_rate = 16000

    def windowed_scoring(self, audios, score_rate):
        from pypesq import pesq
        if len(audios) != 2:
            raise ValueError('NB_PESQ needs a reference and a test signals.')
        return pesq(audios[1], audios[0], score_rate)