test_utils.py 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. import numpy as np
  2. from vedo.utils import make3d
  3. print('----------------------------------8')
  4. print(make3d([]))
  5. assert str(make3d([])) == '[]'
  6. print('----------------------------------9')
  7. print(make3d([0,1]))
  8. assert str(make3d([0,1])) == '[0 1 0]'
  9. print('----------------------------------11')
  10. print(make3d([[0,1],[9,8]]))
  11. assert str(make3d([[0,1],[9,8]])) == '[[0 1 0]\n [9 8 0]]'
  12. print('----------------------------------7')
  13. print(make3d([[0,1], [6,7], [6,7], [6,7]]))
  14. assert str(make3d([[0,1], [6,7], [6,7], [6,7]])) == '[[0 1 0]\n [6 7 0]\n [6 7 0]\n [6 7 0]]'
  15. print('----------------------------------10')
  16. print(make3d([0,1,2]))
  17. assert str(make3d([0,1,2])) == '[0 1 2]'
  18. print('----------------------------------4')
  19. print(make3d([[0,1,2]]))
  20. assert str(make3d([[0,1,2]])) == '[[0 1 2]]'
  21. print('----------------------------------5')
  22. print(make3d([[0,1,2], [6,7,8]]))
  23. assert str(make3d([[0,1,2], [6,7,8]])) == '[[0 1 2]\n [6 7 8]]'
  24. print('----------------------------------3')
  25. print(make3d([ [0,1,2], [6,7,8], [6,7,9] ]))
  26. assert str(make3d([ [0,1,2], [6,7,8], [6,7,9] ])) == '[[0 1 2]\n [6 7 8]\n [6 7 9]]'
  27. print('----------------------------------6')
  28. print(make3d([[0,1,2], [6,7,8], [6,7,8], [6,7,4]]))
  29. assert str(make3d([[0,1,2], [6,7,8], [6,7,8], [6,7,4]])) == '[[0 1 2]\n [6 7 8]\n [6 7 8]\n [6 7 4]]'
  30. # print(make3d([[0,1,2,3], [6,7,8,9]])# will CORRECTLY raise error)
  31. # print(make3d([ [0,1,2,3], [6,7,8,9], [6,7,8,8] ]))# will CORRECTLY raise error
  32. # print(make3d([0,1,2,3])) # will CORRECTLY raise error