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)