Add and improve tests

Author: lanzagar

Orange/tests/test_basket_reader.py

@@ -89,6 +89,10 @@ def test_handles_quote(self, fname):
         table = read_basket(fname)
         self.assertEqual(len(table.domain.variables), 4)
 
+    def test_data_name(self):
+        filename = os.path.join(os.path.dirname(__file__), 'iris_basket.basket')
+        self.assertEqual(read_basket(filename).name, 'iris_basket')
+
 
 if __name__ == "__main__":
     unittest.main()

Orange/tests/test_tab_reader.py

@@ -9,9 +9,10 @@
 
 import numpy as np
 
-from Orange.data import Table, ContinuousVariable, DiscreteVariable
+from Orange.data import Table, DiscreteVariable
 from Orange.data.io import TabReader
 
+
 def read_tab_file(filename):
     return TabReader(filename).read()
 
@@ -152,3 +153,9 @@ def test_attributes_saving(self):
         table = Table(path.join(tempdir, "out.tab"))
         self.assertEqual(table.attributes[1], "test")
         shutil.rmtree(tempdir)
+
+    def test_data_name(self):
+        table1 = Table('iris')
+        table2 = TabReader(table1.__file__).read()
+        self.assertEqual(table1.name, 'iris')
+        self.assertEqual(table2.name, 'iris')

Orange/tests/test_xlsx_reader.py

@@ -6,7 +6,7 @@
 
 import numpy as np
 
-from Orange.data import io, ContinuousVariable, DiscreteVariable, StringVariable
+from Orange.data import io, ContinuousVariable, DiscreteVariable
 
 
 def get_dataset(name):
@@ -31,6 +31,7 @@ def test_read(self):
                                        np.array([[0.1, 0.5, 0.1, 21],
                                                  [0.2, 0.1, 2.5, 123],
                                                  [0, 0, 0, 0]]))
+        self.assertEqual(table.name, 'header_0')
 
 
 class TextExcelSheets(unittest.TestCase):
@@ -45,6 +46,7 @@ def test_named_sheet(self):
         self.reader.select_sheet("my_sheet")
         table = self.reader.read()
         self.assertEqual(len(table.domain.attributes), 4)
+        self.assertEqual(table.name, 'header_0_sheet-my_sheet')
 
 
 class TestExcelHeader1(unittest.TestCase):
@@ -94,6 +96,7 @@ def test_flags(self):
         np.testing.assert_almost_equal(
             table.metas[:, 1], np.array([0, 1, 2, 3] * 5 + [0, 1, 2]))
 
+
 class TestExcelHeader3(unittest.TestCase):
     def test_read(self):
         table = read_file("header_3.xlsx")
@@ -130,5 +133,6 @@ def test_read(self):
         np.testing.assert_equal(
             table.metas[:, 2], np.array(list("abcdefghijklmnopqrstuvw")))
 
+
 if __name__ == "__main__":
     unittest.main()