o jg@sxddlmZddlmZmZmZddlmZddZddZ dd Z d d Z d d Z ddZ ddZddZddZdS)) zip_longest) list_visitorMultisetPartitionTraversermultiset_partitions_taocp)_set_partitionsccs0|D]}|\}}}||kr||kr|VqdS)a Filters (on the number of parts) a multiset partition enumeration Arguments ========= lb, and ub are a range (in the Python slice sense) on the lpart variable returned from a multiset partition enumeration. Recall that lpart is 0-based (it points to the topmost part on the part stack), so if you want to return parts of sizes 2,3,4,5 you would use lb=1 and ub=5. N)partition_iteratorlbubstateflpartpstackrr^/var/www/html/zoom/venv/lib/python3.10/site-packages/sympy/utilities/tests/test_enumerative.pypart_range_filters rc Csg}t||D] \}}||g|qt}t|}t|D]/\}}ddt|D} t|D] } | || || q0ttdd| D} | | q|S)aEnumerates partitions of a multiset Parameters ========== multiplicities list of integer multiplicities of the components of the multiset. components the components (elements) themselves Returns ======= Set of partitions. Each partition is tuple of parts, and each part is a tuple of components (with repeats to indicate multiplicity) Notes ===== Multiset partitions can be created as equivalence classes of set partitions, and this function does just that. This approach is slow and memory intensive compared to the more advanced algorithms available, but the code is simple and easy to understand. Hence this routine is strictly for testing -- to provide a straightforward baseline against which to regress the production versions. (This code is a simplified version of an earlier production implementation.) cSsg|]}gqSrr).0irrr Isz0multiset_partitions_baseline..cSg|]}t|qSrtuplerprrrrM) zipextendsetlenrrangeappendrsortedadd) multiplicities componentscanonctelemcachenncqrvr canonicalrrrmultiset_partitions_baseline!s   r-cCsXd}t||}t}t|D]}ttddt||D}||q||ks*JdS)z~ Enumerates the partitions of multiset with AOCP algorithm and baseline implementation, and compare the results. abcdefghijklmnopqrstuvwxyzcSrrrrrrrrbrz/compare_multiset_w_baseline..N)r-rrrr rr!)r"letters bl_partitionsaocp_partitionsr p1rrrcompare_multiset_w_baselineRs   r3cCsl|\}}}|\}}}||kr4|d|d|d|dkr4|d||d|d||dkr4dSdS)zcompare for equality two instances of multiset partition states This is useful for comparing different versions of the algorithm to verify correctness.rTFr)s1s2f1lpart1pstack1f2lpart2pstack2rrrcompare_multiset_statesgs  ((r=cCs$ddg}t|gd}t|dS)zjCompares the output of multiset_partitions_taocp with a baseline (set partition based) implementation.)r4N)r3)r"rrrtest_multiset_partitions_taocpvs rAcCs>gd}t}t||t|D] \}}t||sJqdS)z4Compares Knuth-based versions of multiset_partitions)r>r>r4N)rrenum_allrr=)r"mr5r6rrr!test_multiset_partitions_versionss  rEcCst}||||ksJt}t}t}||||}tt|||}t||||t|} t|||d|} t ||| | D]\} } } }t | | sQJt | | sXJt | |s_JqDdS)zCompare filter-based and more optimized subrange implementations Helper for tests, called with both small and larger multisets. rN) rcount_partitionscount_partitions_slow enum_rangerr enum_smallsum enum_largerr=)multr r rDmamcmda_itb_itc_itd_itsasbscsdrrrsubrange_exercises rXcC gd}d}d}t|||dS)N)r?r?r>r4r4r>rXrLr r rrr test_subranger\cCrY)N)r@r>r4r?rZr[rrrtest_subrange_larger]r`N) itertoolsrsympy.utilities.enumerativerrrsympy.utilities.iterablesrrr-r3r=rArErXr\r`rrrrs  1