5
5
"""
6
6
import unittest
7
7
8
+ from automatminer import MatPipe
8
9
from automatminer .presets import get_preset_config
9
10
10
11
AF_KEY = "autofeaturizer"
@@ -19,31 +20,43 @@ def test_production(self):
19
20
prod = get_preset_config ("production" )
20
21
for k in KEYSET :
21
22
self .assertTrue (k in prod .keys ())
23
+ MatPipe (** prod )
22
24
23
25
def test_debug (self ):
24
26
debug = get_preset_config ("debug" )
25
27
for k in KEYSET :
26
28
self .assertTrue (k in debug .keys ())
29
+ MatPipe (** debug )
27
30
28
31
def test_debug_single (self ):
29
32
debug_single = get_preset_config ("debug_single" )
30
33
for k in KEYSET :
31
34
self .assertTrue (k in debug_single .keys ())
35
+ MatPipe (** debug_single )
32
36
33
37
def test_express (self ):
34
38
express = get_preset_config ("express" )
35
39
for k in KEYSET :
36
40
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 )
37
48
38
49
def test_heavy (self ):
39
50
heavy = get_preset_config ("heavy" )
40
51
for k in KEYSET :
41
52
self .assertTrue (k in heavy .keys ())
53
+ MatPipe (** heavy )
42
54
43
55
def test_caching_powerup (self ):
44
56
cache_src = "./somefile.json"
45
57
prod = get_preset_config ("production" , cache_src = cache_src )
46
58
self .assertEqual (prod [AF_KEY ].cache_src , cache_src )
59
+ MatPipe (** prod )
47
60
48
61
def test_missing (self ):
49
62
with self .assertRaises (ValueError ):
0 commit comments