Python suite示例

示例#1

文件： __init__.py 项目： ricky/supysonic

def suite():
    suite = unittest.TestSuite()

    suite.addTest(managers.suite())
    suite.addTest(base.suite())
    suite.addTest(api.suite())
    suite.addTest(frontend.suite())

    return suite

示例#2

文件： __init__.py 项目： andrenam/supysonic

def suite():
    suite = unittest.TestSuite()

    suite.addTest(managers.suite())
    suite.addTest(base.suite())
    suite.addTest(api.suite())
    suite.addTest(frontend.suite())

    return suite

示例#3

文件： test_default_forward_pass.py 项目： zhipingzhou/DurIAN

import sys
sys.path.insert(0, '..')

import unittest
import json
import numpy as np

import torch

from model import DurIAN
from base import suite, BaseModelForwardPassTest

seed = 0
np.random.seed(seed)
torch.manual_seed(seed)
torch.backends.cudnn.deterministic = True


class DurIANForwardPassTest(BaseModelForwardPassTest):
    def __init__(self, *args, **kwargs):
        super(DurIANForwardPassTest, self).__init__(*args, **kwargs)
        self.CLASS_TYPE = DurIAN
        with open('../configs/default.json') as f:
            self.config = json.load(f)
        self.config['n_symbols'] = 100


if __name__ == '__main__':
    runner = unittest.TextTestRunner()
    runner.run(suite([DurIANForwardPassTest]))

示例#4

文件： test_baseline_backward_pass.py 项目： zhipingzhou/DurIAN

    def __init__(self, *args, **kwargs):
        super(BaselineDurIANBackwardPassBCETest, self).__init__(*args, **kwargs)
        self.CLASS_TYPE = BaselineDurIAN
        with open('../configs/baseline.json') as f:
            self.config = json.load(f)
        self.config['n_symbols'] = 100
        self.config['solve_alignments_as_mse'] = False
        self.config['solve_alignments_as_bce'] = True
        self.criterion = DurIANLoss(self.config)


class BaselineDurIANBackwardPassJoinTest(BaseModelBackwardPassTest):
    def __init__(self, *args, **kwargs):
        super(BaselineDurIANBackwardPassJoinTest, self).__init__(*args, **kwargs)
        self.CLASS_TYPE = BaselineDurIAN
        with open('../configs/baseline.json') as f:
            self.config = json.load(f)
        self.config['n_symbols'] = 100
        self.config['solve_alignments_as_mse'] = True
        self.config['solve_alignments_as_bce'] = True
        self.criterion = DurIANLoss(self.config)


if __name__ == '__main__':
    runner = unittest.TextTestRunner()
    runner.run(suite([
        BaselineDurIANBackwardPassMSETest,
        BaselineDurIANBackwardPassBCETest,
        BaselineDurIANBackwardPassJoinTest
    ]))

示例#5

文件： test_default_backward_pass.py 项目： zhipingzhou/DurIAN

    def __init__(self, *args, **kwargs):
        super(DurIANBackwardPassBCETest, self).__init__(*args, **kwargs)
        self.CLASS_TYPE = DurIAN
        with open('../configs/default.json') as f:
            self.config = json.load(f)
        self.config['n_symbols'] = 100
        self.config['solve_alignments_as_mse'] = False
        self.config['solve_alignments_as_bce'] = True
        self.criterion = DurIANLoss(self.config)


class DurIANBackwardPassJoinTest(BaseModelBackwardPassTest):
    def __init__(self, *args, **kwargs):
        super(DurIANBackwardPassJoinTest, self).__init__(*args, **kwargs)
        self.CLASS_TYPE = DurIAN
        with open('../configs/default.json') as f:
            self.config = json.load(f)
        self.config['n_symbols'] = 100
        self.config['solve_alignments_as_mse'] = True
        self.config['solve_alignments_as_bce'] = True
        self.criterion = DurIANLoss(self.config)


if __name__ == '__main__':
    runner = unittest.TextTestRunner()
    runner.run(suite([
        DurIANBackwardPassMSETest,
        DurIANBackwardPassBCETest,
        DurIANBackwardPassJoinTest
    ]))

示例#6

import sys
sys.path.insert(0, '..')

import unittest
import json
import numpy as np

import torch

from model import BaselineDurIAN
from base import suite, BaseModelForwardPassTest

seed = 0
np.random.seed(seed)
torch.manual_seed(seed)
torch.backends.cudnn.deterministic = True


class BaselineDurIANForwardPassTest(BaseModelForwardPassTest):
    def __init__(self, *args, **kwargs):
        super(BaselineDurIANForwardPassTest, self).__init__(*args, **kwargs)
        self.CLASS_TYPE = BaselineDurIAN
        with open('../configs/baseline.json') as f:
            self.config = json.load(f)
        self.config['n_symbols'] = 100


if __name__ == '__main__':
    runner = unittest.TextTestRunner()
    runner.run(suite([BaselineDurIANForwardPassTest]))