Skip to content

Commit 8c75c01

Browse files
committed
add a test to make sure instantiation is ok
1 parent 71ae44b commit 8c75c01

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed

automatminer/tests/test_presets.py

+13
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
"""
66
import unittest
77

8+
from automatminer import MatPipe
89
from automatminer.presets import get_preset_config
910

1011
AF_KEY = "autofeaturizer"
@@ -19,31 +20,43 @@ def test_production(self):
1920
prod = get_preset_config("production")
2021
for k in KEYSET:
2122
self.assertTrue(k in prod.keys())
23+
MatPipe(**prod)
2224

2325
def test_debug(self):
2426
debug = get_preset_config("debug")
2527
for k in KEYSET:
2628
self.assertTrue(k in debug.keys())
29+
MatPipe(**debug)
2730

2831
def test_debug_single(self):
2932
debug_single = get_preset_config("debug_single")
3033
for k in KEYSET:
3134
self.assertTrue(k in debug_single.keys())
35+
MatPipe(**debug_single)
3236

3337
def test_express(self):
3438
express = get_preset_config("express")
3539
for k in KEYSET:
3640
self.assertTrue(k in express.keys())
41+
MatPipe(**express)
42+
43+
def test_express_single(self):
44+
express_single = get_preset_config("express_single")
45+
for k in KEYSET:
46+
self.assertTrue(k in express_single.keys())
47+
MatPipe(**express_single)
3748

3849
def test_heavy(self):
3950
heavy = get_preset_config("heavy")
4051
for k in KEYSET:
4152
self.assertTrue(k in heavy.keys())
53+
MatPipe(**heavy)
4254

4355
def test_caching_powerup(self):
4456
cache_src = "./somefile.json"
4557
prod = get_preset_config("production", cache_src=cache_src)
4658
self.assertEqual(prod[AF_KEY].cache_src, cache_src)
59+
MatPipe(**prod)
4760

4861
def test_missing(self):
4962
with self.assertRaises(ValueError):

0 commit comments

Comments
 (0)