i*!ddlZddlZddlZddlmZmZmZmZddlZ ddl m Z ddl m Z ddlmZddlmZmZddlmZmZddlmZdd lmZmZdd lmZdd lmZdd lm Z m!Z!ejDe#Z$d e%dejfdZ&dedeege'fdeee(ffdZ)dedeege'fdeee(ffdZ*defdZ+defdZ,e,ddfde%dededede%dee%defdZ-dde,dddfde%dee%deeded ee d!eed"ee defd#Z.d$ee!d%e%ddfd&Z/ddggfd'e d(e(d)e(d*ee%d+ee%ddf d,Z0d'e d-e%ddfd.Z1d'e d-e%ddfd/Z2d'e ddfd0Z3y)1N)CallableOptionalSequenceTuple)PydanticDimension)PydanticEntity)PydanticMeasure)PydanticFileSlicePydanticMetadata)PydanticMetricPydanticMetricTypeParams)PydanticSemanticManifestPydanticNodeRelationPydanticSemanticModel)YamlConfigFile) MetricType)!SemanticManifestValidationResultsValidationIssue date_stringreturnc@tjj|S)zDHelper to convert a string like '2020-01-01' into a datetime object.)dateutilparserparse)rs c/var/www/html/strategist-ai/venv/lib/python3.12/site-packages/dbt_semantic_interfaces/test_utils.py as_datetimer$s ??  --modelfunctioncnt|jD]\}}||s||fcStd)adReturns a semantic model from the model which matches the criteria defined by the passed in function'. This is useful because the order of semantic models in the list is non determinant, thus it's impossible to hard code which semantic model you want by index. Using semantic model names isn't great for consistency because semantic models change and might no longer have the necessary parts to be useful for a given test. This allows us to guarantee that a semantic model will be returned which meets the requirements of what a test needs, unless none of the semantic models will work. z:Unable to find a semantic_model matching function criteria) enumeratesemantic_models Exception)rr indexsemantic_models rfind_semantic_model_withr')sD"+5+@+@!A)~ N #!5( () P QQrcnt|jD]\}}||s||fcStd)a,Returns a metric from the model which matches the criteria defined by the passed in function'. This is useful because the order of metrics in the list is non-determinant, thus it's impossible to hard code which metric you want by index. Using metric names isn't great for consistency because metrics change and might no longer have the necessary parts to be useful for a given test. This allows us to guarantee that a metric will be returned which meets the requirements of what a test needs, unless none of the metrics will work. z2Unable to find a metric matching function criteria)r"metricsr$)rr r%metrics rfind_metric_withr+;s@#5==1! v F 5= ! H IIrcFtjd}td|S)zReturns a YamlConfigFile with the inputs for a basic valid semantic manifest. This is useful to seed a simple error-free semantic manifest, which can easily be extended with YAML inputs containing specific validation triggers. ag semantic_model: name: sample_semantic_model node_relation: schema_name: some_schema alias: source_table entities: - name: example_entity type: primary role: test_role expr: example_id measures: - name: num_sample_rows agg: sum agg_time_dimension: ds expr: 1 create_metric: true dimensions: - name: ds type: time type_params: time_granularity: day inline_for_test)filepathcontents)textwrapdedentr) yaml_contentss rbase_semantic_manifest_filer3Ms' OO M2 #4} MMrc 6tdtddddS)z8Returns a Metadata object with the required information.z/not/from/a/repoznot_from_file.pyzN/Ar)filenamecontentstart_line_numberend_line_number)repo_file_path file_slice)r r rr default_metar<os& )$'   rz adhoc metricnametype type_paramsmetadata descriptiontime_granularityc &t||||d||S)zyCreates a metric with the given input. If a metadata object is not supplied, a default metadata object is used. N)r=rAr>r?filterr@rB)r )r=r>r?r@rArBs rmetric_with_guaranteed_metarE|s&   ) rr; node_relationentitiesmeasures dimensionsc H|}| tdd}t|||||||S)zCreates a semantic model with the given input. If a metadata object is not supplied, a default metadata object is used. schematable) schema_namealias)r=rArFrGrHrIr@r)r=rArFr@rGrHrIcreated_node_relations r#semantic_model_with_guaranteed_metarPsE*$ 4 !  ! + rissuesmessage_fragmentct|Dcgc] }|jj|dk7"c}}||Dcgc]}|jc}dr|sJycc}wcc}w)N)expectedactual_messages)anymessagefind)rQrRissue found_matchs r#_assert_expected_validation_messager\sivVeu}}))*:;rAVWK%7=>eEMM>     W?s %AAresultsnum_expected_errorsnum_expected_warningsexpected_error_msgsexpected_warning_msgsc"t|j|k(sJt|j|k(sJt|jdk(sJd|D]}t |j||D]}t |j|y)zValidates the number, type, and content of ValidationIssues. Currently assumes zero future_errors as there are no future_errors implemented, but this function can be expanded to cover those if needed. rzAvalidation function expects zero future_errors to be implemented.)rQrRN)lenwarningserrors future_errorsr\)r]r^r_r`raexpected_error_msgexpected_warning_msgs rcheck_expected_issuesris w $9 99 9 w~~ "5 55 5 w$$ % *o,oo *1h+7>>Tfgh 5l+73C3CVjklrtarget_messagec"t|d|gy)N)r]r^r`rir]rjs r!check_only_one_error_with_messageros+,rc"t|d|gy)Nrl)r]r_rarmrns r#check_only_one_warning_with_messagerqs-.rct|y)Nr]rmrss rcheck_no_errors_or_warningsrts r)4datetimeloggingr0typingrrrrdateutil.parserr:dbt_semantic_interfaces.implementations.elements.dimensionr7dbt_semantic_interfaces.implementations.elements.entityr8dbt_semantic_interfaces.implementations.elements.measurer 0dbt_semantic_interfaces.implementations.metadatar r .dbt_semantic_interfaces.implementations.metricr r 9dbt_semantic_interfaces.implementations.semantic_manifestr6dbt_semantic_interfaces.implementations.semantic_modelrr'dbt_semantic_interfaces.parsing.objectsr"dbt_semantic_interfaces.type_enumsr5dbt_semantic_interfaces.validations.validator_helpersrr getLogger__name__loggerstrrboolintr'r+r3r<rErPr\rirorqrtr;rrrs66XRTC9   8 $.S.X%6%6. R #R/79N8OQU8U/VR # %&R$J #J/78H$8N/OJ >3 J$N^ND & "".%&*   *    sm 2"&48!-)+*,.0 #01  ~&  ' *+>  _ %     !!")++- l .lll"# l $C= l  l, .@C  .@C )Jtr