Ejemplos de base_test_version en Python

Ejemplo n.º 1

Archivo: test_multi_turn_dialog.py Proyecto: xiaoanshi/cotk

	def test_version(self):
		base_test_version(SwitchboardCorpus)

Ejemplo n.º 2

Archivo: test_language_generation.py Proyecto: xiaoanshi/cotk

 def test_version(self):
     base_test_version(MSCOCO)

Ejemplo n.º 3

Archivo: test_multi_turn_dialog.py Proyecto: xiaoanshi/cotk

	def test_version(self):
		base_test_version(UbuntuCorpus)

Ejemplo n.º 4

Archivo: test_single_turn_dialog.py Proyecto: yyht/cotk

    def test_init(self, load_opensubtitles):
        super().base_test_init(load_opensubtitles())
        super().base_test_all_unknown(load_opensubtitles(10000))

    def test_restart(self, load_opensubtitles):
        super().base_test_restart(load_opensubtitles())

    @pytest.mark.dependency(depends=["TestOpenSubtitles::test_init"])
    def test_get_batch(self, load_opensubtitles):
        super().base_test_get_batch(load_opensubtitles())

    @pytest.mark.dependency(depends=["TestOpenSubtitles::test_init"])
    def test_get_next_batch(self, load_opensubtitles):
        super().base_test_get_next_batch(load_opensubtitles())

    @pytest.mark.dependency(depends=["TestOpenSubtitles::test_init"])
    def test_convert(self, load_opensubtitles):
        super().base_test_convert(load_opensubtitles())

    def test_teacher_forcing_metric(self, load_opensubtitles):
        super().base_test_teacher_forcing_metric(load_opensubtitles())

    def test_teacher_inference_metric(self, load_opensubtitles):
        super().base_test_teacher_inference_metric(load_opensubtitles())

    def test_init_multi_runs(self, load_opensubtitles):
        super().base_test_multi_runs([load_opensubtitles() for i in range(3)])


base_test_version(OpenSubtitles)

Ejemplo n.º 5

Archivo: test_language_generation.py Proyecto: yyht/cotk

	def test_init(self, load_mscoco):
		super().base_test_init(load_mscoco())
		super().base_test_all_unknown(load_mscoco(10000))

	def test_restart(self, load_mscoco):
		super().base_test_restart(load_mscoco())

	@pytest.mark.dependency(depends=["TestMSCOCO::test_init"])
	def test_get_batch(self, load_mscoco):
		super().base_test_get_batch(load_mscoco())

	@pytest.mark.dependency(depends=["TestMSCOCO::test_init"])
	def test_get_next_batch(self, load_mscoco):
		super().base_test_get_next_batch(load_mscoco())

	@pytest.mark.dependency(depends=["TestMSCOCO::test_init"])
	def test_convert(self, load_mscoco):
		super().base_test_convert(load_mscoco())

	def test_teacher_forcing_metric(self, load_mscoco):
		super().base_test_teacher_forcing_metric(load_mscoco())

	def test_teacher_inference_metric(self, load_mscoco):
		super().base_test_teacher_inference_metric(load_mscoco())

	def test_init_multi_runs(self, load_mscoco):
		super().base_test_multi_runs([load_mscoco() for i in range(3)])


base_test_version(MSCOCO)

Ejemplo n.º 6

        super().base_test_get_next_batch(load_switchboardcorpus())

    @pytest.mark.dependency(depends=["TestSwitchboardCorpus::test_init"])
    def test_convert(self, load_switchboardcorpus):
        super().base_test_convert(load_switchboardcorpus())

    def test_multi_turn_convert(self, load_switchboardcorpus):
        super().base_test_multi_turn_convert(load_switchboardcorpus())

    def test_teacher_forcing_metric(self, load_switchboardcorpus):
        super().base_test_teacher_forcing_metric(load_switchboardcorpus())

    def test_teacher_inference_metric(self, load_switchboardcorpus):
        super().base_test_teacher_inference_metric(load_switchboardcorpus())

    def test_teacher_precision_recall_metric(self, load_switchboardcorpus):
        dl = load_switchboardcorpus()
        glove = Glove("./tests/wordvector/dummy_glove/300d/")
        embed = glove.load_dict(dl.vocab_list)
        assert isinstance(
            dl.get_multi_ref_metric(generated_num_per_context=3,
                                    word2vec=embed), MetricBase)

    def test_init_multi_runs(self, load_switchboardcorpus):
        super().base_test_multi_runs(
            [load_switchboardcorpus() for i in range(3)])


base_test_version(UbuntuCorpus)
base_test_version(SwitchboardCorpus)

Ejemplo n.º 7

Archivo: test_single_turn_dialog.py Proyecto: xiaoanshi/cotk

 def test_version(self):
     base_test_version(OpenSubtitles)

Ejemplo n.º 8

Archivo: test_sentence_classification.py Proyecto: yyht/cotk

class TestSST(TestSentenceClassification):

	@pytest.mark.dependency()
	def test_init(self, load_sst):
		super().base_test_init(load_sst())
		super().base_test_all_unknown(load_sst(1000000))

	def test_restart(self, load_sst):
		super().base_test_restart(load_sst())

	@pytest.mark.dependency(depends=["TestSST::test_init"])
	def test_get_batch(self, load_sst):
		super().base_test_get_batch(load_sst())

	@pytest.mark.dependency(depends=["TestSST::test_init"])
	def test_get_next_batch(self, load_sst):
		super().base_test_get_next_batch(load_sst())

	@pytest.mark.dependency(depends=["TestSST::test_init"])
	def test_convert(self, load_sst):
		super().base_test_convert(load_sst())

	def test_metric(self, load_sst):
		super().base_test_metric(load_sst())

	def test_init_multi_runs(self, load_sst):
		super().base_test_multi_runs([load_sst() for i in range(3)])


base_test_version(SST)