def test_parse_headers_3(self):
hdata = self.header3
headers, data = DataTableMixin.parse_headers(hdata)
self.assertListEqual(
headers, [["a", "b", "c", "d", "w", "e", "f", "g"],
["d", "c", "c", "c", "c", "d", "s", "yes no"],
["meta", "class", "meta", "", "weight", "i", "", ""]])
self.assertListEqual(list(data), hdata[3:])
def test_adjust_data_width_empty(self):
names, types, flags = ["a", "b"], [], []
header = Mock()
header.names = names
header.types = types
header.flags = flags
data = [["", ""], ["", ""]]
adjusted, n = DataTableMixin.adjust_data_width(data, header)
np.testing.assert_array_equal(adjusted, [])
self.assertEqual(n, 2)
self.assertListEqual(names, ["a", "b"])
self.assertListEqual(types, [])
self.assertListEqual(flags, [])
def test_adjust_data_width_shorten(self):
names, types, flags = ["a", "b", "c"], ["", "c"], ["m", "c"]
header = Mock()
header.names = names
header.types = types
header.flags = flags
adjusted, n = DataTableMixin.adjust_data_width(self.header0, header)
np.testing.assert_array_equal(
adjusted, np.array(self.header0, dtype=object)[:, :3])
self.assertEqual(adjusted.shape, (len(self.header0), 3))
self.assertEqual(n, 3)
self.assertListEqual(names, ["a", "b", "c"])
self.assertListEqual(types, ["", "c", ""])
self.assertListEqual(flags, ["m", "c", ""])
def test_adjust_data_width_lengthen(self):
names, types, flags = ["a", "b", "c", "d", "e"], ["", "c"], ["m", "c"]
header = Mock()
header.names = names
header.types = types
header.flags = flags
adjusted, n = DataTableMixin.adjust_data_width(self.header0, header)
_data = np.hstack((np.array(self.header0, dtype=object),
np.array([[""]] * 3, dtype=object)))
np.testing.assert_array_equal(_data, adjusted)
self.assertEqual(adjusted.shape, (len(self.header0), 5))
self.assertEqual(n, 5)
self.assertListEqual(names, ["a", "b", "c", "d", "e"])
self.assertListEqual(types, ["", "c", "", "", ""])
self.assertListEqual(flags, ["m", "c", "", "", ""])
def test_parse_headers_1_flags(self):
hdata = self.header1_flags
headers, data = DataTableMixin.parse_headers(hdata)
self.assertListEqual(headers,
[["m#a", "cC#b", "m#c", "d", "i#e", "f"]])
self.assertListEqual(list(data), hdata[1:])
def test_parse_headers_1(self):
hdata = self.header1
headers, data = DataTableMixin.parse_headers(hdata)
self.assertListEqual(headers, [["a", "b", "c", "d"]])
self.assertListEqual(list(data), hdata[1:])
def test_parse_headers_0(self):
hdata = self.header0
headers, data = DataTableMixin.parse_headers(hdata)
self.assertListEqual(headers, [])
self.assertListEqual(list(data), hdata)
def test_parse_headers_empty(self):
headers, data = DataTableMixin.parse_headers([])
self.assertListEqual(headers, [])
self.assertListEqual(list(data), [])
def test_data_table_3(self):
self.assertIsInstance(DataTableMixin.data_table(self.header3), Table)
def test_data_table_1_flags(self):
self.assertIsInstance(DataTableMixin.data_table(self.header1_flags),
Table)
def test_data_table_empty(self):
self.assertIsInstance(DataTableMixin.data_table([]), Table)
