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))
