Skip to content

Commit 36c9892

Browse files
committed
added missing test modules
1 parent c24b598 commit 36c9892

File tree

2 files changed

+38
-0
lines changed

2 files changed

+38
-0
lines changed

tests/arch.py

+15
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
# Licensed under the Apache License, Version 2.0 (the "License");
2+
# you may not use this file except in compliance with the License.
3+
# You may obtain a copy of the License at
4+
#
5+
# http://www.apache.org/licenses/LICENSE-2.0
6+
#
7+
# Unless required by applicable law or agreed to in writing, software
8+
# distributed under the License is distributed on an "AS IS" BASIS,
9+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
10+
# See the License for the specific language governing permissions and
11+
# limitations under the License.
12+
13+
import platform
14+
15+
apple_silicon = (platform.system() == "Darwin" and platform.processor() == "arm")

tests/predictor_classes.py

+23
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
import mhctools
2+
from .arch import apple_silicon
3+
4+
mhc1_predictor_classes = [
5+
mhctools.NetMHCpan,
6+
mhctools.NetMHC,
7+
mhctools.NetMHC4,
8+
mhctools.IedbNetMHCcons,
9+
mhctools.IedbNetMHCpan,
10+
mhctools.IedbSMM,
11+
mhctools.IedbSMM_PMBEC,
12+
]
13+
14+
if not apple_silicon:
15+
mhc1_predictor_classes += [
16+
mhctools.NetMHC3,
17+
mhctools.NetMHCcons,
18+
]
19+
20+
mhc2_predictor_classes = [
21+
mhctools.NetMHCIIpan,
22+
mhctools.IedbNetMHCIIpan,
23+
]

0 commit comments

Comments
 (0)