<commit_before>#!/usr/bin/env python

from numpy import all, array

from avena import flip


def test_flip_vertical():
    x = array([[1], [2], [3]])
    assert all(flip._flip_vertical(flip._flip_vertical(x)) == x)
    y = array([[3], [2], [1]])
    assert all(flip._flip_vertical(x) == y)


def test_flip_horizontal():
    x = array([[1, 2, 3]])
    assert all(flip._flip_horizontal(flip._flip_horizontal(x)) == x)
    y = array([[3, 2, 1]])
    assert all(flip._flip_horizontal(x) == y)


if __name__ == '__main__':
    pass
<commit_msg>Use relative imports in unit tests.<commit_after>#!/usr/bin/env python

from numpy import all, array

from .. import flip


def test_flip_vertical():
    x = array([[1], [2], [3]])
    assert all(flip._flip_vertical(flip._flip_vertical(x)) == x)
    y = array([[3], [2], [1]])
    assert all(flip._flip_vertical(x) == y)


def test_flip_horizontal():
    x = array([[1, 2, 3]])
    assert all(flip._flip_horizontal(flip._flip_horizontal(x)) == x)
    y = array([[3, 2, 1]])
    assert all(flip._flip_horizontal(x) == y)


if __name__ == '__main__':
    pass

{'fraction_non_alphanumeric': 0.1814038286235187, 'fraction_numerical': 0.02187784867821331, 'mean_word_length': 3.8370044052863435, 'pattern_counts': {'":': 0, '<': 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': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '2815817', 'n_tokens_mistral': 443, 'n_tokens_neox': 406, 'n_words': 98}