def test_invalid_names(self):
output_mock = StringIO.StringIO()
with self.assertRaises(NameError):
utils.wascii(np.asarray([1, 2, 3, 4]),
output_mock,
'%d',
names=['a', 'b', 'c'])
def test_blank(self):
# test outputting integers
test_array = np.asarray([[1, 2, 3, 4], [5, 6, 7, 8]])
answer = '1 2 3 4\n5 6 7 8\n'
output_mock = StringIO.StringIO()
utils.wascii(test_array, output_mock, '%d', blank=True)
result = output_mock.getvalue()
output_mock.close()
self.assertEqual(result, answer)
def test_floats_padding_format(self):
# test outputting floats
test_array = np.asarray([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])
answer = ' 1.00 2.00 3.00 4.00\n 5.00 6.00 7.00 8.00\n 9.00 10.00 11.00 12.00'
output_mock = StringIO.StringIO()
utils.wascii(test_array, output_mock, '%5.2f')
result = output_mock.getvalue()
output_mock.close()
self.assertEqual(result, answer)
def test_names(self):
# test outputting integers
test_array = np.asarray([[1, 2, 3, 4], [5, 6, 7, 8]])
answer = '# Column Descriptions:\n# 1: a\n# 2: b\n# 3: c\n# 4: d\n1 2 3 4\n5 6 7 8'
output_mock = StringIO.StringIO()
utils.wascii(test_array, output_mock, '%d', names=['a', 'b', 'c', 'd'])
result = output_mock.getvalue()
output_mock.close()
self.assertEqual(result, answer)
def test_variable_format(self):
# test passing a format for each column
test_array = np.asarray([[1, 2, 3.16, 4], [5, 6, 7, 8]])
answer = '1 2.00 3.2 4.00\n5 6.00 7.0 8.00'
output_mock = StringIO.StringIO()
utils.wascii(test_array, output_mock,
['%1d', '%5.2f', '%3.1f', '%4.2f'])
result = output_mock.getvalue()
output_mock.close()
self.assertEqual(result, answer)
def test_header_list(self):
# test outputting floats
test_array = np.asarray([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])
answer = '12345 12345 12345 12345\n67890 67890 67890 67890\n 1.00 2.00 3.00 4.00\n 5.00 6.00 7.00 8.00\n 9.00 10.00 11.00 12.00'
output_mock = StringIO.StringIO()
utils.wascii(
test_array,
output_mock,
'%5.2f',
header=['12345 12345 12345 12345', '67890 67890 67890 67890'])
result = output_mock.getvalue()
output_mock.close()
self.assertEqual(result, answer)
def test_invalid_format(self):
output_mock = StringIO.StringIO()
with self.assertRaises(NameError):
utils.wascii(np.asarray([1, 2, 3, 4]), output_mock,
['%d', '%d', '%d'])
