def test_data_frame(self):
"""Should render data Frame"""
df = pd.DataFrame([{
'a': 1,
'b': 'hello',
'c': True,
'd': date(2016, 9, 9)
}, {
'a': 1,
'b': 'hello',
'c': True,
'd': date(2016, 9, 9)
}, {
'a': 1,
'b': 'hello',
'c': True,
'd': date(2016, 9, 9)
}, {
'a': 1,
'b': 'hello',
'c': True,
'd': date(2016, 9, 9)
}])
result = texts.head(df, 3)
self.assertGreater(len(result), 1)
result = texts.tail(df, 3)
self.assertGreater(len(result), 1)
def test_list(self):
"""Should render list"""
source = [{
'a': 1,
'b': 'hello',
'c': True,
'd': date(2016, 9, 9)
}, {
'a': 1,
'b': 'hello',
'c': True,
'd': date(2016, 9, 9)
}, {
'a': 1,
'b': 'hello',
'c': True,
'd': date(2016, 9, 9)
}, {
'a': 1,
'b': 'hello',
'c': True,
'd': date(2016, 9, 9)
}]
result = texts.head(source, 3)
self.assertGreater(len(result), 1)
result = texts.tail(source, 3)
self.assertGreater(len(result), 1)
def test_tail_mock(self):
"""Should show last 3 lines of mock object"""
target = MagicMock()
target.tail.side_effect = ValueError('FAKE')
target.__len__.return_value = 10
target.__iter__.return_value = range(10)
result = texts.tail(target, 3)
self.assertLess(0, len(result), 'Result should not be empty')
def test_dict(self):
"""Should render dictionary"""
source = {'a': 1, 'b': 'hello', 'c': True, 'd': date(2016, 9, 9)}
result = texts.head(source, 2)
self.assertGreater(len(result), 1)
result = texts.tail(source, 2)
self.assertGreater(len(result), 1)
def test_dict(self):
"""Should render dictionary"""
source = {'a': 1, 'b': 'hello', 'c': True, 'd': date(2016, 9, 9)}
result = texts.head(source, 2)
self.assertGreater(len(result), 1)
result = texts.tail(source, 2)
self.assertGreater(len(result), 1)
def test_array(self):
"""Should render array"""
array = np.ndarray([1, 2, 3, 4, 5, 6])
result = texts.head(array, 2)
self.assertGreater(len(result), 1)
result = texts.tail(array, 2)
self.assertGreater(len(result), 1)
def test_array(self):
"""Should render array"""
array = np.ndarray([1, 2, 3, 4, 5, 6])
result = texts.head(array, 2)
self.assertGreater(len(result), 1)
result = texts.tail(array, 2)
self.assertGreater(len(result), 1)
def test_object(self):
"""Should render object"""
class TestObject(object):
def __str__(self):
return 'a\nb\nc\nd\ne'
result = texts.head(TestObject(), 2)
self.assertGreater(len(result), 1)
result = texts.tail(TestObject(), 2)
self.assertGreater(len(result), 1)
def test_tail_mock(self):
"""Should show last 3 lines of mock object"""
target = MagicMock()
target.tail.side_effect = ValueError('FAKE')
target.__len__.return_value = 10
target.__iter__.return_value = range(10)
result = texts.tail(target, 3)
self.assertLess(
0, len(result),
'Result should not be empty'
)
def test_object(self):
"""Should render object"""
class TestObject(object):
def __str__(self):
return 'a\nb\nc\nd\ne'
result = texts.head(TestObject(), 2)
self.assertGreater(len(result), 1)
result = texts.tail(TestObject(), 2)
self.assertGreater(len(result), 1)
def test_data_frame(self):
"""Should render data Frame"""
df = pd.DataFrame([
{'a': 1, 'b': 'hello', 'c': True, 'd': date(2016, 9, 9)},
{'a': 1, 'b': 'hello', 'c': True, 'd': date(2016, 9, 9)},
{'a': 1, 'b': 'hello', 'c': True, 'd': date(2016, 9, 9)},
{'a': 1, 'b': 'hello', 'c': True, 'd': date(2016, 9, 9)}
])
result = texts.head(df, 3)
self.assertGreater(len(result), 1)
result = texts.tail(df, 3)
self.assertGreater(len(result), 1)
def test_list(self):
"""Should render list"""
source = [
{'a': 1, 'b': 'hello', 'c': True, 'd': date(2016, 9, 9)},
{'a': 1, 'b': 'hello', 'c': True, 'd': date(2016, 9, 9)},
{'a': 1, 'b': 'hello', 'c': True, 'd': date(2016, 9, 9)},
{'a': 1, 'b': 'hello', 'c': True, 'd': date(2016, 9, 9)}
]
result = texts.head(source, 3)
self.assertGreater(len(result), 1)
result = texts.tail(source, 3)
self.assertGreater(len(result), 1)
def tail(source, count: int = 5):
"""
The opposite of the head function. Displays the last *count* elements of
the *source* object.
:param source:
DataFrames will show the last *count* rows of that DataFrame. A list,
tuple or other iterable, will show the last *count* rows. Dictionaries
will show *count* keys from the dictionary, which will be randomly
selected unless you are using an OrderedDict. Strings will show the
last *count* characters.
:param count:
The number of elements to show from the source.
"""
_get_report().append_body(render_texts.tail(source, count=count))
def tail(source, count: int = 5):
"""
The opposite of the head function. Displays the last *count* elements of
the *source* object.
:param source:
DataFrames will show the last *count* rows of that DataFrame. A list,
tuple or other iterable, will show the last *count* rows. Dictionaries
will show *count* keys from the dictionary, which will be randomly
selected unless you are using an OrderedDict. Strings will show the
last *count* characters.
:param count:
The number of elements to show from the source.
"""
r = _get_report()
r.append_body(render_texts.tail(source, count=count))
r.stdout_interceptor.write_source('[ADDED] Tail\n')
def test_tail_string(self):
"""Should show last 3 lines of string"""
source = 'a\nb\nc\nd\ne\nf'
result = texts.tail(source, 3)
self.assertTrue(0 < result.find('d\ne\nf'))
def test_tail_empty_data_frame(self):
"""
Should return an empty string when there are no rows to head.
"""
result = texts.tail(pd.DataFrame([]))
self.assertEqual('', result)
def test_tail_no_rows(self):
"""Should return an empty string when no rows are set for display."""
result = texts.tail(None, 0)
self.assertEqual('', result)
def test_tail_no_rows(self):
"""Should return an empty string when no rows are set for display."""
result = texts.tail(None, 0)
self.assertEqual('', result)
def test_tail_string(self):
"""Should show last 3 lines of string"""
source = 'a\nb\nc\nd\ne\nf'
result = texts.tail(source, 3)
self.assertTrue(0 < result.find('d\ne\nf'))
def test_tail_empty_data_frame(self):
"""
Should return an empty string when there are no rows to head.
"""
result = texts.tail(pd.DataFrame([]))
self.assertEqual('', result)