test_conflict.py 1.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162
  1. from sympy.multipledispatch.conflict import (supercedes, ordering, ambiguities,
  2. ambiguous, super_signature, consistent)
  3. class A: pass
  4. class B(A): pass
  5. class C: pass
  6. def test_supercedes():
  7. assert supercedes([B], [A])
  8. assert supercedes([B, A], [A, A])
  9. assert not supercedes([B, A], [A, B])
  10. assert not supercedes([A], [B])
  11. def test_consistent():
  12. assert consistent([A], [A])
  13. assert consistent([B], [B])
  14. assert not consistent([A], [C])
  15. assert consistent([A, B], [A, B])
  16. assert consistent([B, A], [A, B])
  17. assert not consistent([B, A], [B])
  18. assert not consistent([B, A], [B, C])
  19. def test_super_signature():
  20. assert super_signature([[A]]) == [A]
  21. assert super_signature([[A], [B]]) == [B]
  22. assert super_signature([[A, B], [B, A]]) == [B, B]
  23. assert super_signature([[A, A, B], [A, B, A], [B, A, A]]) == [B, B, B]
  24. def test_ambiguous():
  25. assert not ambiguous([A], [A])
  26. assert not ambiguous([A], [B])
  27. assert not ambiguous([B], [B])
  28. assert not ambiguous([A, B], [B, B])
  29. assert ambiguous([A, B], [B, A])
  30. def test_ambiguities():
  31. signatures = [[A], [B], [A, B], [B, A], [A, C]]
  32. expected = {((A, B), (B, A))}
  33. result = ambiguities(signatures)
  34. assert set(map(frozenset, expected)) == set(map(frozenset, result))
  35. signatures = [[A], [B], [A, B], [B, A], [A, C], [B, B]]
  36. expected = set()
  37. result = ambiguities(signatures)
  38. assert set(map(frozenset, expected)) == set(map(frozenset, result))
  39. def test_ordering():
  40. signatures = [[A, A], [A, B], [B, A], [B, B], [A, C]]
  41. ord = ordering(signatures)
  42. assert ord[0] == (B, B) or ord[0] == (A, C)
  43. assert ord[-1] == (A, A) or ord[-1] == (A, C)
  44. def test_type_mro():
  45. assert super_signature([[object], [type]]) == [type]