Re-enable test_dbapi01 that was disabled by mistake

MonetDB · Dec 14, 2018 · 2979506 · 2979506
1 parent 44e2857
commit 2979506
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_dbapi01.py b/tests/test_dbapi01.py
@@ -3,15 +3,15 @@
 import numpy
 
 
-class MultipleResultSets(object):
+class TestMultipleResultSets(object):
     def test_regular_selection(self, monetdblite_cursor):
         monetdblite_cursor.execute('SELECT * FROM integers')
         monetdblite_cursor.execute('SELECT * FROM integers')
         result = monetdblite_cursor.fetchall()
-        assert result == [[0],[1],[2],[3],[4],[5],[6],[7],[8],[9]], "Incorrect result returned"
+        assert result == [[0], [1], [2], [3], [4], [5], [6], [7], [8], [9], [None]], "Incorrect result returned"
 
     def test_numpy_selection(self, monetdblite_cursor):
         monetdblite_cursor.execute('SELECT * FROM integers')
         monetdblite_cursor.execute('SELECT * FROM integers')
         result = monetdblite_cursor.fetchnumpy()
-        assert str(result) == str({'i': numpy.ma.masked_array(numpy.arange(10))}), "Incorrect result returned"
+        assert str(result) == str({'i': numpy.ma.masked_array(numpy.arange(11), [False]*10 + [True])}), "Incorrect result returned"