6
6
from pymatgen .core import Structure
7
7
8
8
from atomate2 .vasp .flows .matpes import MatPesStaticFlowMaker
9
+ from atomate2 .vasp .jobs .matpes import MatPesGGAStaticMaker
9
10
10
11
11
12
def test_matpes_static_flow_maker (mock_vasp , clean_dir , vasp_test_dir ):
@@ -59,7 +60,9 @@ def test_matpes_static_flow_maker(mock_vasp, clean_dir, vasp_test_dir):
59
60
assert flow [0 ].name == "MatPES GGA static"
60
61
61
62
# Test setting two makers to None
62
- flow_maker = MatPesStaticFlowMaker (static1 = None , static2 = None )
63
+ flow_maker = MatPesStaticFlowMaker (
64
+ static1 = None , static2 = None , static3 = MatPesGGAStaticMaker ()
65
+ )
63
66
flow = flow_maker .make (si_struct )
64
67
assert len (flow ) == 0
65
68
@@ -68,7 +71,9 @@ def test_matpes_static_flow_maker(mock_vasp, clean_dir, vasp_test_dir):
68
71
MatPesStaticFlowMaker (static1 = None , static2 = None , static3 = None )
69
72
70
73
# Test no static3 if structure requires no Hubbard U corrections
71
- flow_maker = MatPesStaticFlowMaker (static1 = None , static2 = None )
74
+ flow_maker = MatPesStaticFlowMaker (
75
+ static1 = None , static2 = None , static3 = MatPesGGAStaticMaker ()
76
+ )
72
77
flow = flow_maker .make (si_struct )
73
78
assert len (flow ) == 0
74
79
assert flow .output == {}
0 commit comments