test_size.py 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. import os
  2. import pytest
  3. from numpy.testing import assert_equal
  4. from . import util
  5. def _path(*a):
  6. return os.path.join(*((os.path.dirname(__file__),) + a))
  7. class TestSizeSumExample(util.F2PyTest):
  8. sources = [_path('src', 'size', 'foo.f90')]
  9. @pytest.mark.slow
  10. def test_all(self):
  11. r = self.module.foo([[]])
  12. assert_equal(r, [0], repr(r))
  13. r = self.module.foo([[1, 2]])
  14. assert_equal(r, [3], repr(r))
  15. r = self.module.foo([[1, 2], [3, 4]])
  16. assert_equal(r, [3, 7], repr(r))
  17. r = self.module.foo([[1, 2], [3, 4], [5, 6]])
  18. assert_equal(r, [3, 7, 11], repr(r))
  19. @pytest.mark.slow
  20. def test_transpose(self):
  21. r = self.module.trans([[]])
  22. assert_equal(r.T, [[]], repr(r))
  23. r = self.module.trans([[1, 2]])
  24. assert_equal(r, [[1], [2]], repr(r))
  25. r = self.module.trans([[1, 2, 3], [4, 5, 6]])
  26. assert_equal(r, [[1, 4], [2, 5], [3, 6]], repr(r))
  27. @pytest.mark.slow
  28. def test_flatten(self):
  29. r = self.module.flatten([[]])
  30. assert_equal(r, [], repr(r))
  31. r = self.module.flatten([[1, 2]])
  32. assert_equal(r, [1, 2], repr(r))
  33. r = self.module.flatten([[1, 2, 3], [4, 5, 6]])
  34. assert_equal(r, [1, 2, 3, 4, 5, 6], repr(r))