mirror of
https://github.com/soxoj/maigret.git
synced 2026-05-07 06:24:35 +00:00
Soxoj
51 lines
955 B
Python
51 lines
955 B
Python
import pytest
|
|
from maigret.permutator import Permute
|
|
|
|
|
|
def test_gather_strict():
|
|
elements = {'a': 1, 'b': 2}
|
|
permute = Permute(elements)
|
|
result = permute.gather(method="strict")
|
|
expected = {
|
|
'a_b': 1,
|
|
'b_a': 2,
|
|
'a-b': 1,
|
|
'b-a': 2,
|
|
'a.b': 1,
|
|
'b.a': 2,
|
|
'ab': 1,
|
|
'ba': 2,
|
|
'_ab': 1,
|
|
'ab_': 1,
|
|
'_ba': 2,
|
|
'ba_': 2,
|
|
}
|
|
assert result == expected
|
|
|
|
|
|
def test_gather_all():
|
|
elements = {'a': 1, 'b': 2}
|
|
permute = Permute(elements)
|
|
result = permute.gather(method="all")
|
|
expected = {
|
|
'a': 1,
|
|
'_a': 1,
|
|
'a_': 1,
|
|
'b': 2,
|
|
'_b': 2,
|
|
'b_': 2,
|
|
'a_b': 1,
|
|
'b_a': 2,
|
|
'a-b': 1,
|
|
'b-a': 2,
|
|
'a.b': 1,
|
|
'b.a': 2,
|
|
'ab': 1,
|
|
'ba': 2,
|
|
'_ab': 1,
|
|
'ab_': 1,
|
|
'_ba': 2,
|
|
'ba_': 2,
|
|
}
|
|
assert result == expected
|