ni*SSKrSSKJr SSKrSSKJr SrSSKr SSK r SSK r SSK r SSKJr Sr\ R"S5r\ R"S5r\ R(R+S /S Q5S 5r\ R(R.\ R(R+S /S Q5S 55r\ R(R.\ R(R+S /S Q5S 55r\ R(R.\ R(R+S /S Q5S55r\ R(R7\(+SS9\ R(R+S /S Q5S55r\ R(R7SSS9\ R(R+S S/5S55r\ R(R+S /S Q5S5rg!\a SrGNf=f)N) ThresholdTF)closedc[R"5 [R"5 [R"5 S[RlUS:XaS[RlgUS:XaS[Rlgg)z3Configure PyReason settings for the specified mode.TfpparallelN)prreset reset_rulesreset_settingssettingsverbose fp_versionparallel_computing)modes a/home/james-whalen/.local/lib/python3.13/site-packages/tests/functional/test_advanced_features.py setup_modersYHHJNNBKK t|!%   )- & cUSSRnUSSRnX#-n[R"US5nUS4$)Nr)lowernpround) annotationsweightsprob_Aprob_B union_probs rprobability_funcrsK ^A  $ $F ^A  $ $FJ*a(J q=rcR[RRUSS/5nU$)z6Head function that returns the input node lists as-is.r)numbatypedList)rresults r identity_funcr%'s)[[  {1~a01 2F Mrr)regularrrc[U5 [RR5nUR [RR[ SS5/55 UR [RR[ SS5/55 [RRSS/5n[ X5n[ RX5nX4:Xdeg)zAEnsure annotation function behaves the same with and without JIT.g{Gz?g?g?N)rr!r"r#appendrrpy_func)rrrjit_respy_ress r!test_probability_func_consistencyr,.st++""$Ku{{''c):(;<=u{{''S)9(:;<kkSz*G{4G  % %k ;F   rc[U5 [R"[5 [R "5nUR SSS9 UR SSS9 URSSSS9 [R"U5 [R"[R"SS55 [R"[R"SS 55 [R"[R"S S 55 [R"[R"S S 55 [R"SS9nUR[R"S5SS9(deUR[R"S5SS9(deUR[R"S5SS9(deUR[R"S5SS9(deg)z:Test head function usage in rules for node and edge rules.Ar)propertyB) connectedzHProcessed(identity_func(X)) <- property(X), property(Y), connected(X, Y)node_rule_with_funczGRoute(identity_func(A), B) <- property(X), property(Y), connected(X, Y)edge_rule_func_firstzFPath(A, identity_func(B)) <- property(X), property(Y), connected(X, Y)edge_rule_func_secondzULink(identity_func(A), identity_func(B)) <- property(X), property(Y), connected(X, Y)edge_rule_func_both timestepsz Processed(A)T) return_boolz Route(A, B)z Path(A, B)z Link(A, B)N)rradd_head_functionr%nxDiGraphadd_nodeadd_edge load_graphadd_ruleRulereasonqueryQuery)rgraphinterpretations rtest_head_functionsrF;sgt' JJLE NN3N# NN3N# NN3qN)MM%KKbdyz{KKacyz{KK`byz{KKorGHIYY+N    8d  KK K    7T  JJ J    6D  II I    6D  II Irc[U5 S[Rl[R"[R "S55 [R"[R "S55 [R "[5 [R"[R"SSS95 [R"SS9n[R"US/5n[U5H(up4[S U35 [U5 [5 M* UR[R"S 55(dS 5eg ) z,Test annotation function usage in reasoning.TzP(A) : [0.01, 1]zP(B) : [0.2, 1]zDunion_probability(A, B):probability_func <- P(A):[0, 1], P(B):[0, 1]) infer_edgesrr6union_probability TIMESTEP - z#union_probability(A, B) : [0.21, 1]z Union probability should be 0.21N)rrr allow_ground_rulesadd_factFactadd_annotation_functionrr?r@rAfilter_and_sort_edges enumerateprintrBrC)rrE dataframestdfs rtest_annotation_functionrUVst%)BKK"KK*+,KK)*+/0KK^lpqrYY+N)).;N:OPJ:& A3  b  '   )N O P PtRtt Prc [U5 SnS[Rl[R"U5 [ SSS5[ SSS5/n[R "[R"SS US 95 [R"[R"S S S S55 [R"[R"SSS S55 [R"[R"SSSS55 [R"[R"SSSS55 [R"SS9n[R"US/5n[U5H(upV[SU35 [U5 [5 M* [US 5S :XdS5e[US5S:XdS5eSUSSR;a"USR S R"SS/:XdS5eg)z$Test custom threshold functionality.z+./tests/functional/group_chat_graph.graphmlT greater_equal)numbertotalr)percentrYdz,ViewedByAll(y) <- HaveAccess(x,y), Viewed(x)viewed_by_all_rule)custom_thresholdsz Viewed(Zach)zseen-fact-zachrrzViewed(Justin)zseen-fact-justinzViewed(Michelle)zseen-fact-michellez Viewed(Amy)z seen-fact-amyr6 ViewedByAllrJz7At t=0 the TextMessage should not have been ViewedByAllz3At t=2 the TextMessage should have been ViewedByAll TextMessage componentzBTextMessage should have ViewedByAll bounds [1,1] for t=2 timestepsN)rrr atom_trace load_graphmlrr?r@rLrMrAfilter_and_sort_nodesrPrQlenvaluesilocr_)r graph_pathuser_defined_thresholdsrErRrSrTs rtest_custom_thresholdsrjnst?J"BKKOOJ /#6:/#7= KK  : 5 KK(8!Q?@KK(*YY+N)).=/JJ:& A3  b  ' JqMaA@A JqMa=<= JqM+6== =*Q-BTBT Ck  CL L L Crztorch not installed)rAc[U5 [R"SS5n/SQn[R"SSSSS9n[R "XS US 9n[ R"S S5nS nS nU"XVU5upn [U[ R5(dS 5e[U 5[U5:XdS5e[SU5 [SU 5 [S5 U Hn [U 5 M g)z*Test classifier integration with PyReason. r)catdograbbitg?TFgffffff?) thresholdset_lower_boundset_upper_bound snap_value classifier) identifierinterface_optionsrrz)The model output should be a torch.TensorzExpected one fact per classLogits Probabilitiesz Generated PyReason Facts:N) rnnLinearrModelInterfaceOptionsLogicIntegratedClassifiertorchrand isinstanceTensorrerQ) rmodel class_namesrvlogic_classifier input_tensort1t2output probabilitiesfactsfacts rtest_classifier_integrationrst IIb! E+K 00 33ES_FWY::a$L B B$4Lb#I F5 fell + +X-XX + u:[) )H+HH ) (F /=) '( d rz9Reason again functionality not implemented for FP versionr&c[U5 Sn[R"U5 [R"[R"SS55 [R "[R "SSSS55 [R5 [R"SS9n[R "SS S S 5n[R "U5 [R"S S SS9n[R"US/5n[U5H(upV[SU35 [U5 [5 M* [US 5S:XdS5e[US 5S :XdS5e[US 5S :XdS5eSUS SR;a"US RSR SS/:XdS5eSUS SR;a"US RSR SS/:XdS5eSUS SR;a"US RSR SS/:XdS5eSUS SR;a"US RSR SS/:XdS5eSUS SR;a"US RS R SS/:XdS5eSUS SR;a"US RSR SS/:XdS5eg) z*Test reasoning continuation functionality.z(./tests/functional/friends_graph.graphmlz=popular(x) <-1 popular(y), Friends(x,y), owns(y,z), owns(x,z) popular_rulez popular(Mary) popular_factrrr6 popular_fact2r^rTF)r7againrestartpopularrJz)At t=0 there should be one popular personz)At t=1 there should be two popular peoplez+At t=2 there should be three popular peopleMaryraz7Mary should have popular bounds [1,1] for t=0 timestepsz7Mary should have popular bounds [1,1] for t=1 timestepsz7Mary should have popular bounds [1,1] for t=2 timestepsJustinz9Justin should have popular bounds [1,1] for t=1 timestepsz9Justin should have popular bounds [1,1] for t=2 timestepsJohnz7John should have popular bounds [1,1] for t=2 timestepsN)rrrcr?r@rLrM faulthandlerenablerArdrPrQrerfrgr)rrhrEnew_factrRrSrTs rtest_reason_againrs+t?YY+NwwA>HKKYY$FN)).9+FJ:& A3  b  ' z!}  "O$OO " z!}  "O$OO " z!}  "Q$QQ " Z];/66 6:a=;M;Ma;P;X;X]^`a\b;b^e^^ b Z];/66 6:a=;M;Ma;P;X;X]^`a\b;b^e^^ b Z];/66 6:a=;M;Ma;P;X;X]^`a\b;b^e^^ b z!}[188 8Z]=O=OPQ=R=Z=Z_`bc^d=dbgbb d z!}[188 8Z]=O=OPQ=R=Z=Z_`bc^d=dbgbb d Z];/66 6:a=;M;Ma;P;X;X]^`a\b;b^e^^ b;brc[U5 [R"5nURSS/5 [R "U5 [R "[R"SS55 [R "[R"SS55 [R"[R"SS55 [R"S 5nS [Rl [R"S U/S 9ng )z.Test reasoning with query-based rule filtering)r.r0)r0Czpopular(x) <-1 friend(x, y)rule1zfriend(x, y) <-1 knows(x, y)rule2z knows(A, B)fact1z popular(A)Fr)r7queriesN)rr:r;add_edges_fromrr>r?r@rLrMrCr r rA)rrDrBrEs rtest_reason_with_queriesr st JJLE *j12MM%KK5w?@KK6@AKK w/0 HH\ "EBKKYYUGrs O  K .   !>? @ !>?J@J2!>?u@u,!>?7L@7Lt'0EF!>?,@G,^D!\])-*^.^*^Z!>?=@=GOs GG'&G'