# -*- coding: utf-8 -*- from __future__ import absolute_import, division, print_function import numpy as np import pandas as pd from xarray.core import formatting from xarray.core.pycompat import PY3 from . import raises_regex class TestFormatting(object): def test_get_indexer_at_least_n_items(self): cases = [ ((20,), (slice(10),), (slice(-10, None),)), ((3, 20,), (0, slice(10)), (-1, slice(-10, None))), ((2, 10,), (0, slice(10)), (-1, slice(-10, None))), ((2, 5,), (slice(2), slice(None)), (slice(-2, None), slice(None))), ((1, 2, 5,), (0, slice(2), slice(None)), (-1, slice(-2, None), slice(None))), ((2, 3, 5,), (0, slice(2), slice(None)), (-1, slice(-2, None), slice(None))), ((1, 10, 1,), (0, slice(10), slice(None)), (-1, slice(-10, None), slice(None))), ((2, 5, 1,), (slice(2), slice(None), slice(None)), (slice(-2, None), slice(None), slice(None))), ((2, 5, 3,), (0, slice(4), slice(None)), (-1, slice(-4, None), slice(None))), ((2, 3, 3,), (slice(2), slice(None), slice(None)), (slice(-2, None), slice(None), slice(None))), ] for shape, start_expected, end_expected in cases: actual = formatting._get_indexer_at_least_n_items(shape, 10, from_end=False) assert start_expected == actual actual = formatting._get_indexer_at_least_n_items(shape, 10, from_end=True) assert end_expected == actual def test_first_n_items(self): array = np.arange(100).reshape(10, 5, 2) for n in [3, 10, 13, 100, 200]: actual = formatting.first_n_items(array, n) expected = array.flat[:n] assert (expected == actual).all() with raises_regex(ValueError, 'at least one item'): formatting.first_n_items(array, 0) def test_last_n_items(self): array = np.arange(100).reshape(10, 5, 2) for n in [3, 10, 13, 100, 200]: actual = formatting.last_n_items(array, n) expected = array.flat[-n:] assert (expected == actual).all() with raises_regex(ValueError, 'at least one item'): formatting.first_n_items(array, 0) def test_last_item(self): array = np.arange(100) reshape = ((10, 10), (1, 100), (2, 2, 5, 5)) expected = np.array([99]) for r in reshape: result = formatting.last_item(array.reshape(r)) assert result == expected def test_format_item(self): cases = [ (pd.Timestamp('2000-01-01T12'), '2000-01-01T12:00:00'), (pd.Timestamp('2000-01-01'), '2000-01-01'), (pd.Timestamp('NaT'), 'NaT'), (pd.Timedelta('10 days 1 hour'), '10 days 01:00:00'), (pd.Timedelta('-3 days'), '-3 days +00:00:00'), (pd.Timedelta('3 hours'), '0 days 03:00:00'), (pd.Timedelta('NaT'), 'NaT'), ('foo', "'foo'"), (u'foo', "'foo'" if PY3 else "u'foo'"), (b'foo', "b'foo'" if PY3 else "'foo'"), (1, '1'), (1.0, '1.0'), ] for item, expected in cases: actual = formatting.format_item(item) assert expected == actual def test_format_items(self): cases = [ (np.arange(4) * np.timedelta64(1, 'D'), '0 days 1 days 2 days 3 days'), (np.arange(4) * np.timedelta64(3, 'h'), '00:00:00 03:00:00 06:00:00 09:00:00'), (np.arange(4) * np.timedelta64(500, 'ms'), '00:00:00 00:00:00.500000 00:00:01 00:00:01.500000'), (pd.to_timedelta(['NaT', '0s', '1s', 'NaT']), 'NaT 00:00:00 00:00:01 NaT'), (pd.to_timedelta(['1 day 1 hour', '1 day', '0 hours']), '1 days 01:00:00 1 days 00:00:00 0 days 00:00:00'), ([1, 2, 3], '1 2 3'), ] for item, expected in cases: actual = ' '.join(formatting.format_items(item)) assert expected == actual def test_format_array_flat(self): actual = formatting.format_array_flat(np.arange(100), 2) expected = '0 ... 99' assert expected == actual actual = formatting.format_array_flat(np.arange(100), 9) expected = '0 ... 99' assert expected == actual actual = formatting.format_array_flat(np.arange(100), 10) expected = '0 1 ... 99' assert expected == actual actual = formatting.format_array_flat(np.arange(100), 13) expected = '0 1 ... 98 99' assert expected == actual actual = formatting.format_array_flat(np.arange(100), 15) expected = '0 1 2 ... 98 99' assert expected == actual actual = formatting.format_array_flat(np.arange(100.0), 11) expected = '0.0 ... 99.0' assert expected == actual actual = formatting.format_array_flat(np.arange(100.0), 1) expected = '0.0 ... 99.0' assert expected == actual actual = formatting.format_array_flat(np.arange(3), 5) expected = '0 1 2' assert expected == actual actual = formatting.format_array_flat(np.arange(4.0), 11) expected = '0.0 ... 3.0' assert expected == actual actual = formatting.format_array_flat(np.arange(0), 0) expected = '' assert expected == actual actual = formatting.format_array_flat(np.arange(1), 0) expected = '0' assert expected == actual actual = formatting.format_array_flat(np.arange(2), 0) expected = '0 1' assert expected == actual actual = formatting.format_array_flat(np.arange(4), 0) expected = '0 ... 3' assert expected == actual def test_pretty_print(self): assert formatting.pretty_print('abcdefghij', 8) == 'abcde...' assert formatting.pretty_print(u'ß', 1) == u'ß' def test_maybe_truncate(self): assert formatting.maybe_truncate(u'ß', 10) == u'ß' def test_format_timestamp_out_of_bounds(self): from datetime import datetime date = datetime(1300, 12, 1) expected = '1300-12-01' result = formatting.format_timestamp(date) assert result == expected date = datetime(2300, 12, 1) expected = '2300-12-01' result = formatting.format_timestamp(date) assert result == expected def test_attribute_repr(self): short = formatting.summarize_attr(u'key', u'Short string') long = formatting.summarize_attr(u'key', 100 * u'Very long string ') newlines = formatting.summarize_attr(u'key', u'\n\n\n') tabs = formatting.summarize_attr(u'key', u'\t\t\t') assert short == ' key: Short string' assert len(long) <= 80 assert long.endswith(u'...') assert u'\n' not in newlines assert u'\t' not in tabs def test_set_numpy_options(): original_options = np.get_printoptions() with formatting.set_numpy_options(threshold=10): assert len(repr(np.arange(500))) < 200 # original options are restored assert np.get_printoptions() == original_options def test_short_array_repr(): cases = [ np.random.randn(500), np.random.randn(20, 20), np.random.randn(5, 10, 15), np.random.randn(5, 10, 15, 3), ] # number of lines: # for default numpy repr: 167, 140, 254, 248 # for short_array_repr: 1, 7, 24, 19 for array in cases: num_lines = formatting.short_array_repr(array).count('\n') + 1 assert num_lines < 30