(issue 574): added new test array 'io_int_labels' (wide only)

alixdamman · alixdamman · commit 6a9e5dcd0966 · 2018-02-22T15:30:27.000+01:00
diff --git a/larray/tests/data/test.xlsx b/larray/tests/data/test.xlsx
diff --git a/larray/tests/data/testint_labels.csv b/larray/tests/data/testint_labels.csv
@@ -0,0 +1,10 @@
+a,b\c,0,1,2
+0,0,0,1,2
+0,1,3,4,5
+0,2,6,7,8
+1,0,9,10,11
+1,1,12,13,14
+1,2,15,16,17
+2,0,18,19,20
+2,1,21,22,23
+2,2,24,25,26
diff --git a/larray/tests/test_array.py b/larray/tests/test_array.py
@@ -91,6 +91,7 @@ def setUp(self):
         self.io_1d = ndtest(3)
         self.io_2d = ndtest("a=1..3; b=b0,b1")
         self.io_3d = ndtest("a=1..3; b=b0,b1; c=c0..c2")
+        self.io_int_labels = ndtest("a=0..2; b=0..2; c=0..2")
         self.io_unsorted = ndtest("a=3..1; b=b1,b0; c=c2..c0")
         self.io_missing_values = ndtest("a=1..3; b=b0,b1; c=c0..c2", dtype=float)
         self.io_missing_values[2, 'b0'] = np.nan
@@ -2608,6 +2609,9 @@ def test_read_csv(self):
         res = read_csv(inputpath('test3d.csv'))
         assert_array_equal(res, self.io_3d)
 
+        res = read_csv(inputpath('testint_labels.csv'))
+        assert_array_equal(res, self.io_int_labels)
+
         res = read_csv(inputpath('test2d_classic.csv'))
         assert_array_equal(res, ndtest("a=a0..a2; b0..b2"))
 
@@ -2675,6 +2679,9 @@ def test_read_excel_xlwings(self):
         arr = read_excel(inputpath('test.xlsx'), '3d')
         assert_array_equal(arr, self.io_3d)
 
+        arr = read_excel(inputpath('test.xlsx'), 'int_labels')
+        assert_array_equal(arr, self.io_int_labels)
+
         arr = read_excel(inputpath('test.xlsx'), '2d_classic')
         assert_array_equal(arr, ndtest("a=a0..a2; b0..b2"))
 
@@ -2756,6 +2763,9 @@ def test_read_excel_pandas(self):
         arr = read_excel(inputpath('test.xlsx'), '3d', engine='xlrd')
         assert_array_equal(arr, self.io_3d)
 
+        arr = read_excel(inputpath('test.xlsx'), 'int_labels', engine='xlrd')
+        assert_array_equal(arr, self.io_int_labels)
+
         arr = read_excel(inputpath('test.xlsx'), '2d_classic', engine='xlrd')
         assert_array_equal(arr, ndtest("a=a0..a2; b0..b2"))
 
