<commit_before>from __future__ import division, absolute_import, print_function

import os

from numpy import array
from numpy.testing import run_module_suite, assert_equal, dec
from . import util


def _path(*a):
    return os.path.join(*((os.path.dirname(__file__),) + a))


class TestSizeSumExample(util.F2PyTest):
    sources = [_path('src', 'size', 'foo.f90')]

    @dec.slow
    def test_all(self):
        r = self.module.foo([[]])
        assert_equal(r, [0], repr(r))

        r = self.module.foo([[1, 2]])
        assert_equal(r, [3], repr(r))

        r = self.module.foo([[1, 2], [3, 4]])
        assert_equal(r, [3, 7], repr(r))

        r = self.module.foo([[1, 2], [3, 4], [5, 6]])
        assert_equal(r, [3, 7, 11], repr(r))

    @dec.slow
    def test_transpose(self):
        r = self.module.trans([[]])
        assert_equal(r, array([[]]).T, repr(r))

        r = self.module.trans([[1, 2]])
        assert_equal(r, [[1], [2]], repr(r))

        r = self.module.trans([[1, 2, 3], [4, 5, 6]])
        assert_equal(r, [[1, 4], [2, 5], [3, 6]], repr(r))

    @dec.slow
    def test_flatten(self):
        r = self.module.flatten([[]])
        assert_equal(r, [], repr(r))

        r = self.module.flatten([[1, 2]])
        assert_equal(r, [1, 2], repr(r))

        r = self.module.flatten([[1, 2, 3], [4, 5, 6]])
        assert_equal(r, [1, 2, 3, 4, 5, 6], repr(r))

if __name__ == "__main__":
    run_module_suite()
<commit_msg>[TST] slightly simplify f2py test<commit_after>from __future__ import division, absolute_import, print_function

import os

from numpy.testing import run_module_suite, assert_equal, dec
from . import util


def _path(*a):
    return os.path.join(*((os.path.dirname(__file__),) + a))


class TestSizeSumExample(util.F2PyTest):
    sources = [_path('src', 'size', 'foo.f90')]

    @dec.slow
    def test_all(self):
        r = self.module.foo([[]])
        assert_equal(r, [0], repr(r))

        r = self.module.foo([[1, 2]])
        assert_equal(r, [3], repr(r))

        r = self.module.foo([[1, 2], [3, 4]])
        assert_equal(r, [3, 7], repr(r))

        r = self.module.foo([[1, 2], [3, 4], [5, 6]])
        assert_equal(r, [3, 7, 11], repr(r))

    @dec.slow
    def test_transpose(self):
        r = self.module.trans([[]])
        assert_equal(r.T, [[]], repr(r))

        r = self.module.trans([[1, 2]])
        assert_equal(r, [[1], [2]], repr(r))

        r = self.module.trans([[1, 2, 3], [4, 5, 6]])
        assert_equal(r, [[1, 4], [2, 5], [3, 6]], repr(r))

    @dec.slow
    def test_flatten(self):
        r = self.module.flatten([[]])
        assert_equal(r, [], repr(r))

        r = self.module.flatten([[1, 2]])
        assert_equal(r, [1, 2], repr(r))

        r = self.module.flatten([[1, 2, 3], [4, 5, 6]])
        assert_equal(r, [1, 2, 3, 4, 5, 6], repr(r))

if __name__ == "__main__":
    run_module_suite()

{'fraction_non_alphanumeric': 0.20756681707740368, 'fraction_numerical': 0.03852828878861506, 'mean_word_length': 2.6901408450704225, 'pattern_counts': {'":': 2, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 4, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '6412317', 'n_tokens_mistral': 1211, 'n_tokens_neox': 1112, 'n_words': 259}