NgRdZddlmZddlZddlZddlZddlZddlm Z m Z m Z ddl m Z mZmZmZmZmZmZmZddlmZmZmZmZddlmZddlmZmZd Zd?dZ d@dZ!dAdZ"dBdZ#dBdZ$dCdZ% dDdEd&Z& dFdGd,Z' dDdHd.Z( dDdId/Z) dJdKd1Z*dFdLd2Z+ dDdMd4Z, dDdNd8Z-dOd:Z.dPd<Z/dQd>Z0dS)Rz!Utilities that ease unit-testing.) annotationsN)AnyListOptional)ANY MagicMockMock PropertyMockcallcreate_autospec mock_openpatch)CaptureFixtureFixtureRequestLogCaptureFixture MonkeyPatch)Element)elements_from_jsonelements_to_json)rrrrrr rr class_mock function_mockinitializer_mock instance_mock method_mock property_mockelements List[Element]returnNonect|}|Jt|}t|}|J||ksJtd||dS)abRaises AssertionError if `elements -> JSON -> List[Element] -> JSON` are not equal. The procedure is: 1. Serialize `elements` to (original) JSON. 2. Deserialize that JSON to `List[Element]`. 3. Serialize that `List[Element]` to JSON. 3. Compare the original and round-tripped JSON, raise if they are different. N)textz JSON differs:)rr_diff)r original_jsonround_tripped_elementsround_tripped_jsons X/var/www/html/ai-engine/env/lib/python3.11/site-packages/test_unstructured/unit_utils.pyassert_round_trips_through_JSONr',sz%X..M  $ $ $/]CCC)*@AA  ) ) )  . . .+]11 . . . . . list[Element]cZt|D]\}}|||S)z5Updates the `id` attribute of each element to a hash.) enumerate id_to_hash)ridxelements r&assign_hash_idsr/Ds9!(++  W3 Or(headingstractualexpectedc|d}|d}d}|dtj||zS)z~Diff of actual compared to expected. "+" indicates unexpected lines actual, "-" indicates lines missing from actual. T)keependszFdiff: '+': unexpected lines in actual, '-': lines missing from actual ) splitlinesjoindifflibDiffercompare)r0r2r3expected_lines actual_liness r&r"r"Ksd (($(77N$$d$33LWG RWWW^--55lNSSTT TTr( file_namectjtjjdz }||z }t |S)zGResolve the absolute-path to `file_name` in the example-docs directory.z example-docs)pathlibPath__file__parentr1resolve)r>example_docs_dir file_paths r&example_doc_pathrGVsA|H--4;nL 9,I y  "" # ##r(ctt|5}|cdddS#1swxYwYdS)z?Contents of example-doc `file_name` as text (decoded as utf-8).N)openrGread)r>fs r&example_doc_textrL]s y)) * *avvxxs>AA datetime_str Optional[str]Optional[dt.datetime]cH|rtj|ndS)zWParse `datetime_str` to a datetime.datetime instance or None if `datetime_str` is None.N)dtdatetime fromisoformat)rMs r&parse_optional_datetimerTcs"6B L2; $ $\ 2 2 2Lr(Trequestr q_class_nameautospecboolkwargsrr c |t|fd|i|}||j|S)a$Return mock patching class with qualified name `q_class_name`. The mock is autospec'ed based on the patched class unless the optional argument `autospec` is set to False. Any other keyword arguments are passed through to Mock(). Patch is reversed after calling test returns. rWr addfinalizerstopstart)rUrVrWrY_patchs r&rrpsB< = =( =f = =F %%% <<>>r(clstype attr_namename str | Nonec ||jn|}tj||fd|i|}||j|S)zcReturn a mock for attribute `attr_name` on `cls`. Patch is reversed after pytest uses it. Nrc) fixturenamerobjectr\r]r^)rUr`rbrcrYr_s r& cls_attr_mockrh~sW#',7  DD \#y > >t >v > >F %%% <<>>r(q_function_namec |t|fd|i|}||j|S)z|Return mock patching function with qualified name `q_function_name`. Patch is reversed after calling test returns. rWr[)rUrirWrYr_s r&rrsB? @ @X @ @ @F %%% <<>>r(c tj|df|dd|}||j|S)zaReturn mock for __init__() method on `cls`. The patch is reversed after pytest uses it. __init__N)rW return_valuerrgr\r]r^)rUr`rWrYr_s r&rrsI\#z ZH4 Z ZSY Z ZF %%% <<>>r(spec_setc :||n|j}t|f||dd|S)abReturn a mock for an instance of `cls` that draws its spec from the class. The mock does not allow new attributes to be set on the instance. If `name` is missing or |None|, the name of the returned |Mock| instance is set to *request.fixturename*. Additional keyword arguments are passed through to the Mock() call that creates the mock. NT)_nameroinstance)rfr )rUr`rcrorYs r&rrs5#44)>r(std_modtypes.ModuleTypefn_namec tj||fd|i|}||j|S)zfReturn mock for function `fn_name` on `std_mod`. The patch is reversed after pytest uses it. rWrn)rUrxrzrWrYr_s r&stdlib_fn_mockr|sF\'7 H HX H H HF %%% <<>>r( module_namec d|z}t|tfddi|}||j|S)z?Return a mock for the builtin `open()` method in `module_name`.z%s.opencreateT)rr r\r]r^)rUr}rYtargetr_s r& open_mockrsP  $F 69;; > >t >v > >F %%% <<>>r( prop_namec tj||fdti|}||j|S)zgA mock for property `prop_name` on class `cls`. Patch is reversed at the end of the test run. new_callable)rrgr r\r]r^)rUr`rrYr_s r&rrsF \#y N N| Nv N NF %%% <<>>r( q_var_namec xt|fi|}||j|S)z}Return a mock patching the variable with qualified name `q_var_name`. Patch is reversed after calling test returns. r[)rUrrYr_s r&var_mockrs= : ( ( ( (F %%% <<>>r()rrrr)rr)rr))r0r1r2r1r3r1)r>r1rr1)rMrNrrO)T) rUrrVr1rWrXrYrrr )N) rUrr`rarbr1rcrdrYrrr ) rUrrir1rWrXrYrrr ) rUrr`rarWrXrYrrr )NT) rUrr`rarcrdrorXrYrrr )rUrrcrdrYrrr ) rUrr`rarvr1rWrXrYrrr ) rUrrxryrzr1rWrXrYrrr )rUrr}r1rYrrr ) rUrr`rarr1rYrrr )rUrrr1rYrrr )1__doc__ __future__rrRrQr9r@typestypingrrr unittest.mockrrr r r r r rpytestrrrrunstructured.documents.elementsrunstructured.staging.baserr__all__r'r/r"rGrLrTrrhrrrrurr|rrrrtr(r&rs''"""""" &&&&&&&&&&                    RQQQQQQQQQQQ333333JJJJJJJJ $0UUUU$$$$ MMMMBF     $ $EI     :>      XXXXX"%%%%%     (      r(