def test_write_file(self): fd, filename = mkstemp(suffix=".xlsx") os.close(fd) data = Table("zoo") io.ExcelReader.write_file(filename, data, with_annotations=True) reader = io.ExcelReader(filename) read_data = reader.read() domain1 = data.domain domain2 = read_data.domain self.assertEqual(len(domain1.attributes), len(domain2.attributes)) self.assertEqual(len(domain1.class_vars), len(domain2.class_vars)) self.assertEqual(len(domain1.metas), len(domain2.metas)) for var1, var2 in zip(domain1.variables + domain1.metas, domain2.variables + domain2.metas): self.assertEqual(type(var1), type(var2)) self.assertEqual(var1.name, var2.name) if var1.is_discrete: self.assertEqual(var1.values, var2.values) np.testing.assert_array_equal(data.X, read_data.X) np.testing.assert_array_equal(data.Y, read_data.Y) np.testing.assert_array_equal(data.metas, read_data.metas) np.testing.assert_array_equal(data.W, read_data.W) os.unlink(filename)
def setUp(self): self.reader = io.ExcelReader(get_dataset("header_0_sheet.xlsx"))
def read_file(name): return io.ExcelReader(get_dataset(name)).read()
def get_xlsx_reader(name: str) -> io.ExcelReader: return io.ExcelReader(get_dataset(name))
def read_file(name): return io.ExcelReader().read_file( os.path.join(os.path.dirname(__file__), "xlsx_files", name))