Export all names from wine. More testing.

luispedro · May 22, 2009 · 81a2c6a · 81a2c6a
1 parent 798e216
commit 81a2c6a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/milksets/wine/__init__.py b/milksets/wine/__init__.py
@@ -19,6 +19,6 @@
 #  OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 #  THE SOFTWARE.
 
-from .wine import load
+from .wine import *
 
 # vim: set ts=4 sts=4 sw=4 expandtab smartindent:
diff --git a/tests/test_classification_loaders.py b/tests/test_classification_loaders.py
@@ -2,11 +2,12 @@
 import milksets.yeast
 
 def test_classification_loaders():
-    def test_loader(loader):
-        features,labels = loader()
+    def test_loader(module):
+        features,labels = module.load()
         assert len(features) == len(labels)
         assert labels.min() == 0
+        assert labels.max() == len(module.label_names) - 1
         assert (features.std(0)**2).sum() > 1e-8
-    yield test_loader, milksets.wine.load
-    yield test_loader, milksets.yeast.load
+    yield test_loader, milksets.wine
+    yield test_loader, milksets.yeast