1
+ import pytest
1
2
import spacy
2
3
3
4
from zshot import PipelineConfig
7
8
FineTunedLMExtensionStrategy , SummarizationStrategy , ParaphrasingStrategy , EntropyHeuristic
8
9
9
10
11
+ @pytest .mark .skip (reason = "Too expensive to run on every commit" )
10
12
def test_pretrained_lm_extension_strategy ():
11
13
description = "The name of a company"
12
14
strategy = PreTrainedLMExtensionStrategy ()
@@ -19,6 +21,7 @@ def test_pretrained_lm_extension_strategy():
19
21
assert len (desc_variations ) == 3 and len (set (desc_variations + [description ])) == 4
20
22
21
23
24
+ @pytest .mark .skip (reason = "Too expensive to run on every commit" )
22
25
def test_finetuned_lm_extension_strategy ():
23
26
description = "The name of a company"
24
27
strategy = FineTunedLMExtensionStrategy ()
@@ -31,6 +34,7 @@ def test_finetuned_lm_extension_strategy():
31
34
assert len (desc_variations ) == 3 and len (set (desc_variations + [description ])) == 4
32
35
33
36
37
+ @pytest .mark .skip (reason = "Too expensive to run on every commit" )
34
38
def test_summarization_strategy ():
35
39
description = "The name of a company"
36
40
strategy = SummarizationStrategy ()
@@ -43,6 +47,7 @@ def test_summarization_strategy():
43
47
assert len (desc_variations ) == 3 and len (set (desc_variations + [description ])) == 4
44
48
45
49
50
+ @pytest .mark .skip (reason = "Too expensive to run on every commit" )
46
51
def test_paraphrasing_strategy ():
47
52
description = "The name of a company"
48
53
strategy = ParaphrasingStrategy ()
@@ -55,6 +60,7 @@ def test_paraphrasing_strategy():
55
60
assert len (desc_variations ) == 3 and len (set (desc_variations + [description ])) == 4
56
61
57
62
63
+ @pytest .mark .skip (reason = "Too expensive to run on every commit" )
58
64
def test_entropy_heuristic ():
59
65
def check_is_tuple (x ):
60
66
return isinstance (x , tuple ) and len (x ) == 2 and isinstance (x [0 ], str ) and isinstance (x [1 ], float )
0 commit comments