@@ -125,8 +125,8 @@ def test_constructor_defaults(self):
125125 self .assertIs (document .client , client )
126126 self .assertEqual (document .content , content )
127127 self .assertIsNone (document .gcs_url )
128+ self .assertIsNone (document .language )
128129 self .assertEqual (document .doc_type , MUT .Document .PLAIN_TEXT )
129- self .assertEqual (document .language , MUT .DEFAULT_LANGUAGE )
130130 self .assertEqual (document .encoding , MUT .Encoding .UTF8 )
131131
132132 def test_constructor_explicit (self ):
@@ -146,6 +146,13 @@ def test_constructor_explicit(self):
146146 self .assertEqual (document .language , language )
147147 self .assertEqual (document .encoding , MUT .Encoding .UTF32 )
148148
149+ def test_constructor_explicit_language (self ):
150+ client = object ()
151+ content = 'abc'
152+ document = self ._make_one (client , content , language = 'en-US' )
153+ self .assertEqual (document .language , 'en-US' )
154+ self .assertEqual (document ._to_dict ()['language' ], 'en-US' )
155+
149156 def test_constructor_no_text (self ):
150157 with self .assertRaises (ValueError ):
151158 self ._make_one (None , content = None , gcs_url = None )
@@ -162,7 +169,6 @@ def test__to_dict_with_content(self):
162169 info = document ._to_dict ()
163170 self .assertEqual (info , {
164171 'content' : content ,
165- 'language' : document .language ,
166172 'type' : klass .PLAIN_TEXT ,
167173 })
168174
@@ -173,7 +179,6 @@ def test__to_dict_with_gcs(self):
173179 info = document ._to_dict ()
174180 self .assertEqual (info , {
175181 'gcsContentUri' : gcs_url ,
176- 'language' : document .language ,
177182 'type' : klass .PLAIN_TEXT ,
178183 })
179184
@@ -183,7 +188,6 @@ def test__to_dict_with_no_content(self):
183188 document .content = None # Manually unset the content.
184189 info = document ._to_dict ()
185190 self .assertEqual (info , {
186- 'language' : document .language ,
187191 'type' : klass .PLAIN_TEXT ,
188192 })
189193
@@ -203,12 +207,10 @@ def _expected_data(content, encoding_type=None,
203207 extract_sentiment = False ,
204208 extract_entities = False ,
205209 extract_syntax = False ):
206- from google .cloud .language .document import DEFAULT_LANGUAGE
207210 from google .cloud .language .document import Document
208211
209212 expected = {
210213 'document' : {
211- 'language' : DEFAULT_LANGUAGE ,
212214 'type' : Document .PLAIN_TEXT ,
213215 'content' : content ,
214216 },
0 commit comments