18
18
19
19
import static com .google .common .truth .Truth .assertThat ;
20
20
21
- import com .google .cloud .language .v1 .LanguageServiceClient ;
22
21
import com .google .cloud .language .v1 .Entity ;
22
+ import com .google .cloud .language .v1 .LanguageServiceClient ;
23
23
import com .google .cloud .language .v1 .PartOfSpeech .Tag ;
24
24
import com .google .cloud .language .v1 .Sentiment ;
25
25
import com .google .cloud .language .v1 .Token ;
@@ -62,14 +62,14 @@ public class AnalyzeIT {
62
62
}
63
63
64
64
@ Test public void analyzeEntities_withEntitiesFile_containsGod () throws Exception {
65
- // Act
66
- List <Entity > entities =
67
- analyzeApp .analyzeEntitiesFile ("gs://" + BUCKET + "/natural-language/gettysburg.txt" );
68
- List <String > got = entities .stream ().map (e -> e .getName ()).collect (Collectors .toList ());
65
+ // Act
66
+ List <Entity > entities =
67
+ analyzeApp .analyzeEntitiesFile ("gs://" + BUCKET + "/natural-language/gettysburg.txt" );
68
+ List <String > got = entities .stream ().map (e -> e .getName ()).collect (Collectors .toList ());
69
69
70
- // Assert
71
- assertThat (got ).named ("entity names" ).contains ("God" );
72
- }
70
+ // Assert
71
+ assertThat (got ).named ("entity names" ).contains ("God" );
72
+ }
73
73
74
74
@ Test public void analyzeSentimentText_returnPositive () throws Exception {
75
75
// Act
@@ -83,15 +83,15 @@ public class AnalyzeIT {
83
83
}
84
84
85
85
@ Test public void analyzeSentimentFile_returnPositiveFile () throws Exception {
86
- // Act
87
- Sentiment sentiment =
88
- analyzeApp .analyzeSentimentFile ("gs://" + BUCKET + "/natural-language/"
89
- + "sentiment/bladerunner-pos.txt" );
86
+ // Act
87
+ Sentiment sentiment =
88
+ analyzeApp .analyzeSentimentFile ("gs://" + BUCKET + "/natural-language/"
89
+ + "sentiment/bladerunner-pos.txt" );
90
90
91
- // Assert
92
- assertThat ((double )sentiment .getMagnitude ()).isGreaterThan (0.0 );
93
- assertThat ((double )sentiment .getScore ()).isGreaterThan (0.0 );
94
- }
91
+ // Assert
92
+ assertThat ((double )sentiment .getMagnitude ()).isGreaterThan (0.0 );
93
+ assertThat ((double )sentiment .getScore ()).isGreaterThan (0.0 );
94
+ }
95
95
96
96
@ Test public void analyzeSentiment_returnNegative () throws Exception {
97
97
// Act
@@ -105,26 +105,26 @@ public class AnalyzeIT {
105
105
}
106
106
107
107
@ Test public void analyzeSentiment_returnNegativeFile () throws Exception {
108
- // Act
109
- Sentiment sentiment =
110
- analyzeApp .analyzeSentimentFile ("gs://" + BUCKET + "/natural-language/"
111
- + "sentiment/bladerunner-neg.txt" );
108
+ // Act
109
+ Sentiment sentiment =
110
+ analyzeApp .analyzeSentimentFile ("gs://" + BUCKET + "/natural-language/"
111
+ + "sentiment/bladerunner-neg.txt" );
112
112
113
- // Assert
114
- assertThat ((double )sentiment .getMagnitude ()).isGreaterThan (0.0 );
115
- assertThat ((double )sentiment .getScore ()).isLessThan (0.0 );
116
- }
113
+ // Assert
114
+ assertThat ((double )sentiment .getMagnitude ()).isGreaterThan (0.0 );
115
+ assertThat ((double )sentiment .getScore ()).isLessThan (0.0 );
116
+ }
117
117
118
118
@ Test public void analyzeSentiment_returnNeutralFile () throws Exception {
119
- // Act
120
- Sentiment sentiment =
121
- analyzeApp .analyzeSentimentFile ("gs://" + BUCKET + "/natural-language/"
122
- + "sentiment/bladerunner-neutral.txt" );
119
+ // Act
120
+ Sentiment sentiment =
121
+ analyzeApp .analyzeSentimentFile ("gs://" + BUCKET + "/natural-language/"
122
+ + "sentiment/bladerunner-neutral.txt" );
123
123
124
- // Assert
125
- assertThat ((double )sentiment .getMagnitude ()).isGreaterThan (1.0 );
126
- assertThat ((double )sentiment .getScore ()).isWithin (0.1 );
127
- }
124
+ // Assert
125
+ assertThat ((double )sentiment .getMagnitude ()).isGreaterThan (1.0 );
126
+ assertThat ((double )sentiment .getScore ()).isGreaterThan (0.0 );
127
+ }
128
128
129
129
@ Test public void analyzeSyntax_partOfSpeech () throws Exception {
130
130
// Act
@@ -141,18 +141,18 @@ public class AnalyzeIT {
141
141
}
142
142
143
143
@ Test public void analyzeSyntax_partOfSpeechFile () throws Exception {
144
- // Act
145
- List <Token > token =
146
- analyzeApp .analyzeSyntaxFile ("gs://" + BUCKET + "/natural-language/"
147
- + "sentiment/bladerunner-neutral.txt" );
148
-
149
- List <Tag > got = token .stream ().map (e -> e .getPartOfSpeech ().getTag ())
150
- .collect (Collectors .toList ());
151
-
152
- // Assert
153
- assertThat (got ).containsExactly (Tag .PRON , Tag .CONJ , Tag .VERB , Tag .CONJ , Tag .VERB ,
154
- Tag .DET , Tag .NOUN , Tag .PUNCT , Tag .NOUN , Tag .VERB , Tag .ADJ , Tag .PUNCT , Tag .CONJ ,
155
- Tag .ADV , Tag .PRON , Tag .VERB , Tag .VERB , Tag .VERB , Tag .ADJ , Tag .PUNCT , Tag .DET ,
156
- Tag .NOUN , Tag .VERB , Tag .ADV , Tag .ADJ ,Tag .PUNCT ).inOrder ();
157
- }
144
+ // Act
145
+ List <Token > token =
146
+ analyzeApp .analyzeSyntaxFile ("gs://" + BUCKET + "/natural-language/"
147
+ + "sentiment/bladerunner-neutral.txt" );
148
+
149
+ List <Tag > got = token .stream ().map (e -> e .getPartOfSpeech ().getTag ())
150
+ .collect (Collectors .toList ());
151
+
152
+ // Assert
153
+ assertThat (got ).containsExactly (Tag .PRON , Tag .CONJ , Tag .VERB , Tag .CONJ , Tag .VERB ,
154
+ Tag .DET , Tag .NOUN , Tag .PUNCT , Tag .NOUN , Tag .VERB , Tag .ADJ , Tag .PUNCT , Tag .CONJ ,
155
+ Tag .ADV , Tag .PRON , Tag .VERB , Tag .VERB , Tag .VERB , Tag .ADJ , Tag .PUNCT , Tag .DET ,
156
+ Tag .NOUN , Tag .VERB , Tag .ADV , Tag .ADJ ,Tag .PUNCT ).inOrder ();
157
+ }
158
158
}
0 commit comments