Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
34 changes: 34 additions & 0 deletions .github/workflows/property_tools_tests.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
name: DeePMD Property Tools Tests

on:
push:
paths:
- "deepmd/deepmd_property_tools/**"
- ".github/workflows/property_tools_tests.yml"
pull_request:
paths:
- "deepmd/deepmd_property_tools/**"
- ".github/workflows/property_tools_tests.yml"

jobs:
test:
runs-on: ubuntu-latest

steps:
- uses: actions/checkout@v4

- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: "3.10"

- name: Install lightweight test dependencies
run: |
python -m pip install --upgrade pip
python -m pip install numpy pytest

- name: Run unit tests
env:
PYTHONPATH: deepmd/deepmd_property_tools
run: |
python -m pytest deepmd/deepmd_property_tools/tests -v
41 changes: 41 additions & 0 deletions deepmd/deepmd_property_tools/DATA/dataset_demo.csv
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
SMILES,Property
O=[N+](C(COCOCC([N+]([O-])=O)([N+]([O-])=O)[N+]([O-])=O)([N+]([O-])=O)[N+]([O-])=O)[O-],468.15
N1(C=NN=N1)CCN(CCN2C=NN=N2)CCN3N=NN=C3,472.15
OCCN1N=C(N(/N=N/C2=NN(CCO)N=N2)/N=N/C3=NN(CCO)N=N3)N=N1,392.15
C1(N(C2=NN=CN=N2)C3=NN=CN=N3)=NN=CN=N1,504.15
O=[N+](C(COC(OCC([N+]([O-])=O)([N+]([O-])=O)[N+]([O-])=O)(OCC([N+]([O-])=O)([N+]([O-])=O)[N+]([O-])=O)OCC([N+]([O-])=O)([N+]([O-])=O)[N+]([O-])=O)([N+]([O-])=O)[N+]([O-])=O)[O-],464.15
O=[N+](C(COC(OCC([N+]([O-])=O)([N+]([O-])=O)[N+]([O-])=O)(OCC([N+]([O-])=O)([N+]([O-])=O)[N+]([O-])=O)OCC([N+]([O-])=O)([N+]([O-])=O)[N+]([O-])=O)([N+]([O-])=O)[N+]([O-])=O)[O-],307.15
O=C(OCC(OC(N[N+]([O-])=O)=O)COC(N[N+]([O-])=O)=O)N[N+]([O-])=O,425.15
O=[N+](OCC(CO[N+]([O-])=O)O[N+]([O-])=O)[O-],470.55
O=[N+](OC(C(C(CO[N+]([O-])=O)O[N+]([O-])=O)O[N+]([O-])=O)CO[N+]([O-])=O)[O-],447.95
O=C(OCC)N(C1=NON=C1N([N+]([O-])=O)C(OCC)=O)[N+]([O-])=O,397.15
O=C(OCCOCCOC(CN=[N+]=[N-])=O)CN=[N+]=[N-],504.52
O=[N+](C1=NON=C1N([N+]([O-])=O)CN([N+]([O-])=O)CN([N+]([O-])=O)C2=NON=C2[N+]([O-])=O)[O-],423.15
CC1(O[N+]([O-])=O)COC1,414.15
CN(CC(N[N+]([O-])=O)=O)[N+]([O-])=O,386.15
CN(CC(O)=O)[N+]([O-])=O,426.15
CN(N)N=O,394.15
CN[N+]([O-])=O,359.15
CN(CC(NNCC([N+]([O-])=O)([N+]([O-])=O)[N+]([O-])=O)=O)[N+]([O-])=O,381.15
CC1=CN(N2C(N)=NN=N2)N=N1,402.85
O=[N+](NC1=NN(CCO[N+]([O-])=O)N=N1)[O-],364.07
CN([N+]([O-])=O)CN1N=C([N+]([O-])=O)C=C1[N+]([O-])=O,483.15
NC1=C(N(CCCN2N=NN=C2N)N=C1[N+]([O-])=O)[N+]([O-])=O,538.15
O=[N+](C([N+]([O-])=O)([N+]([O-])=O)CNC1=NN=NN1OC)[O-],421.15
N/C(NC)=C([N+]([O-])=O)/[N+]([O-])=O,517.61
N/C(NO)=C([N+]([O-])=O)/[N+]([O-])=O,348.15
O=[N+](C(CNN1C=NN=C1)([N+]([O-])=O)[N+]([O-])=O)[O-],399.65
NC1=NN=C2N=NC(NC3=NON=C3[N+]([O-])=O)=NN21,522.65
NC1=NN=C2N=NC(NC3=NC([N+]([O-])=O)=NO3)=NN21,495.15
O=[N+](C1=NNC([N+]([O-])=O)=C1C2=NC(C3=NNC(C4=C([N+]([O-])=O)NN=C4[N+]([O-])=O)=N3)=NN2)[O-],645.15
NC1=NNC(NC2=NN=C(NC3=NC(N)=NN3)N=N2)=N1,630.25
C1(NC2=NN=C(NC3=NC=NN3)N=N2)=NC=NN1,623.15
O=[N+](C1=NNC(NC2=NN=C(NC3=NC([N+]([O-])=O)=NN3)N=N2)=N1)[O-],556.75
O=[N+](C1=NNC(NC2=NN=C(N=N2)NC3=NC([N+]([O-])=O)=NN3)=N1)[O-],575.15
O=[N+](N(CN([N+]([O-])=O)C1=NON=C1C#N)C2=NON=C2C#N)[O-],405.15
[N-]=[N+]=NC1=NNC(NC2=NN=C(NC3=NC(N=[N+]=[N-])=NN3)N=N2)=N1,462.95
CN1N=NC(NC2=NN=C(NC3=NN(C)N=N3)N=N2)=N1,579.15
C12=NN=CN1N=CN=N2,516.15
O=[N+](C1=NN(C([N+]([O-])=O)=C1[N+]([O-])=O)CN2C([N+]([O-])=O)=C([N+]([O-])=O)C([N+]([O-])=O)=N2)[O-],478.15
NC(N=C1N)=NN1C2=NN=C(N3C(N)=NC(N)=N3)N=N2,643.15
NC1=NC([N+]([O-])=O)=NN1NCC([N+]([O-])=O)([N+]([O-])=O)[N+]([O-])=O,381.15
68 changes: 68 additions & 0 deletions deepmd/deepmd_property_tools/DATA/mol_convert/id0.mol
Original file line number Diff line number Diff line change
@@ -0,0 +1,68 @@
id_0
RDKit 3D

31 30 0 0 0 0 0 0 0 0999 V2000
-4.0868 -2.2052 0.2024 O 0 0 0 0 0 0 0 0 0 0 0 0
-3.7794 -1.1379 0.7719 N 0 0 0 0 0 0 0 0 0 0 0 0
-3.5750 0.0975 0.0395 C 0 0 0 0 0 0 0 0 0 0 0 0
-2.1664 0.6026 0.2558 C 0 0 0 0 0 0 0 0 0 0 0 0
-1.1878 -0.2919 -0.1677 O 0 0 0 0 0 0 0 0 0 0 0 0
0.0296 0.3329 0.1059 C 0 0 0 0 0 0 0 0 0 0 0 0
1.1188 -0.4204 -0.2538 O 0 0 0 0 0 0 0 0 0 0 0 0
2.2495 0.3227 0.0753 C 0 0 0 0 0 0 0 0 0 0 0 0
3.5404 -0.3978 -0.2717 C 0 0 0 0 0 0 0 0 0 0 0 0
4.6077 0.5165 0.1368 N 0 0 0 0 0 0 0 0 0 0 0 0
4.7075 0.9066 1.4645 O 0 0 0 0 0 0 0 0 0 0 0 0
5.4450 0.9674 -0.6709 O 0 0 0 0 0 0 0 0 0 0 0 0
3.6406 -1.6072 0.5234 N 0 0 0 0 0 0 0 0 0 0 0 0
4.7676 -2.3971 0.3350 O 0 0 0 0 0 0 0 0 0 0 0 0
2.7577 -1.9409 1.3499 O 0 0 0 0 0 0 0 0 0 0 0 0
3.6480 -0.6610 -1.6710 N 0 0 0 0 0 0 0 0 0 0 0 0
3.6392 0.4035 -2.5383 O 0 0 0 0 0 0 0 0 0 0 0 0
3.7453 -1.8348 -2.0532 O 0 0 0 0 0 0 0 0 0 0 0 0
-3.8054 -0.1104 -1.3649 N 0 0 0 0 0 0 0 0 0 0 0 0
-5.0305 -0.5411 -1.8063 O 0 0 0 0 0 0 0 0 0 0 0 0
-2.9036 0.0961 -2.1897 O 0 0 0 0 0 0 0 0 0 0 0 0
-4.4563 1.1242 0.5864 N 0 0 0 0 0 0 0 0 0 0 0 0
-4.3995 2.3807 0.0074 O 0 0 0 0 0 0 0 0 0 0 0 0
-5.2364 0.8918 1.5395 O 0 0 0 0 0 0 0 0 0 0 0 0
-3.6336 -1.1517 2.1410 O 0 0 0 0 0 0 0 0 0 0 0 0
-2.0458 1.6118 -0.1858 H 0 0 0 0 0 0 0 0 0 0 0 0
-1.9931 0.7205 1.3611 H 0 0 0 0 0 0 0 0 0 0 0 0
0.0031 0.5823 1.1860 H 0 0 0 0 0 0 0 0 0 0 0 0
-0.0190 1.3102 -0.4505 H 0 0 0 0 0 0 0 0 0 0 0 0
2.2324 0.5226 1.1653 H 0 0 0 0 0 0 0 0 0 0 0 0
2.1862 1.3075 -0.4292 H 0 0 0 0 0 0 0 0 0 0 0 0
1 2 2 0
2 3 1 0
3 4 1 0
4 5 1 0
5 6 1 0
6 7 1 0
7 8 1 0
8 9 1 0
9 10 1 0
10 11 1 0
10 12 2 0
9 13 1 0
13 14 1 0
13 15 2 0
9 16 1 0
16 17 1 0
16 18 2 0
3 19 1 0
19 20 1 0
19 21 2 0
3 22 1 0
22 23 1 0
22 24 2 0
2 25 1 0
4 26 1 0
4 27 1 0
6 28 1 0
6 29 1 0
8 30 1 0
8 31 1 0
M CHG 8 2 1 10 1 11 -1 13 1 14 -1 16 1 17 -1 19 1
M CHG 4 20 -1 22 1 23 -1 25 -1
M END
81 changes: 81 additions & 0 deletions deepmd/deepmd_property_tools/DATA/mol_convert/id1.mol
Original file line number Diff line number Diff line change
@@ -0,0 +1,81 @@
id_1
RDKit 3D

37 39 0 0 0 0 0 0 0 0999 V2000
3.8222 -0.1814 -0.3417 N 0 0 0 0 0 0 0 0 0 0 0 0
4.5522 -0.3657 0.7531 C 0 0 0 0 0 0 0 0 0 0 0 0
5.7306 0.2803 0.5756 N 0 0 0 0 0 0 0 0 0 0 0 0
5.7113 0.8392 -0.6036 N 0 0 0 0 0 0 0 0 0 0 0 0
4.5579 0.5697 -1.1791 N 0 0 0 0 0 0 0 0 0 0 0 0
2.4597 -0.6831 -0.6040 C 0 0 0 0 0 0 0 0 0 0 0 0
1.5795 0.4509 -0.1062 C 0 0 0 0 0 0 0 0 0 0 0 0
0.1709 0.2021 -0.2382 N 0 0 0 0 0 0 0 0 0 0 0 0
-0.5289 1.3503 -0.0115 C 0 0 0 0 0 0 0 0 0 0 0 0
-1.9891 1.4177 0.2362 C 0 0 0 0 0 0 0 0 0 0 0 0
-2.3144 2.8669 0.3392 N 0 0 0 0 0 0 0 0 0 0 0 0
-2.6762 3.6111 -0.7120 C 0 0 0 0 0 0 0 0 0 0 0 0
-2.8832 4.8731 -0.2581 N 0 0 0 0 0 0 0 0 0 0 0 0
-2.6507 4.8934 1.0457 N 0 0 0 0 0 0 0 0 0 0 0 0
-2.3021 3.6409 1.3803 N 0 0 0 0 0 0 0 0 0 0 0 0
-0.1843 -1.1091 0.0202 C 0 0 0 0 0 0 0 0 0 0 0 0
-1.6335 -1.5036 -0.1927 C 0 0 0 0 0 0 0 0 0 0 0 0
-1.8525 -2.9336 0.0557 N 0 0 0 0 0 0 0 0 0 0 0 0
-1.7115 -3.8633 -0.9035 N 0 0 0 0 0 0 0 0 0 0 0 0
-1.9817 -5.0724 -0.4726 N 0 0 0 0 0 0 0 0 0 0 0 0
-2.3172 -4.9296 0.8413 N 0 0 0 0 0 0 0 0 0 0 0 0
-2.2245 -3.6138 1.1112 C 0 0 0 0 0 0 0 0 0 0 0 0
4.2641 -0.9390 1.6390 H 0 0 0 0 0 0 0 0 0 0 0 0
2.3462 -1.6067 -0.0661 H 0 0 0 0 0 0 0 0 0 0 0 0
2.3786 -0.7589 -1.7085 H 0 0 0 0 0 0 0 0 0 0 0 0
1.8944 1.3712 -0.6355 H 0 0 0 0 0 0 0 0 0 0 0 0
1.8574 0.6131 0.9733 H 0 0 0 0 0 0 0 0 0 0 0 0
-0.0187 1.9592 0.7992 H 0 0 0 0 0 0 0 0 0 0 0 0
-0.3141 2.0454 -0.9260 H 0 0 0 0 0 0 0 0 0 0 0 0
-2.6508 1.0137 -0.4999 H 0 0 0 0 0 0 0 0 0 0 0 0
-2.1934 1.0735 1.2958 H 0 0 0 0 0 0 0 0 0 0 0 0
-2.7735 3.2608 -1.7092 H 0 0 0 0 0 0 0 0 0 0 0 0
0.1310 -1.4703 1.0524 H 0 0 0 0 0 0 0 0 0 0 0 0
0.3872 -1.8310 -0.6477 H 0 0 0 0 0 0 0 0 0 0 0 0
-2.2162 -1.0229 0.6248 H 0 0 0 0 0 0 0 0 0 0 0 0
-1.9931 -1.2926 -1.2078 H 0 0 0 0 0 0 0 0 0 0 0 0
-2.4339 -3.1555 2.0821 H 0 0 0 0 0 0 0 0 0 0 0 0
1 2 1 0
2 3 2 0
3 4 1 0
4 5 2 0
1 6 1 0
6 7 1 0
7 8 1 0
8 9 1 0
9 10 1 0
10 11 1 0
11 12 1 0
12 13 2 0
13 14 1 0
14 15 2 0
8 16 1 0
16 17 1 0
17 18 1 0
18 19 1 0
19 20 2 0
20 21 1 0
21 22 2 0
5 1 1 0
15 11 1 0
22 18 1 0
2 23 1 0
6 24 1 0
6 25 1 0
7 26 1 0
7 27 1 0
9 28 1 0
9 29 1 0
10 30 1 0
10 31 1 0
12 32 1 0
16 33 1 0
16 34 1 0
17 35 1 0
17 36 1 0
22 37 1 0
M END
67 changes: 67 additions & 0 deletions deepmd/deepmd_property_tools/DATA/mol_convert/id10.mol
Original file line number Diff line number Diff line change
@@ -0,0 +1,67 @@
id_10
RDKit 3D

31 30 0 0 0 0 0 0 0 0999 V2000
2.8426 -1.6192 -0.4119 O 0 0 0 0 0 0 0 0 0 0 0 0
2.7892 -0.5803 0.3112 C 0 0 0 0 0 0 0 0 0 0 0 0
2.6274 0.6586 -0.2997 O 0 0 0 0 0 0 0 0 0 0 0 0
2.5198 0.8167 -1.6862 C 0 0 0 0 0 0 0 0 0 0 0 0
1.3445 0.0841 -2.2872 C 0 0 0 0 0 0 0 0 0 0 0 0
0.1663 0.5630 -1.7163 O 0 0 0 0 0 0 0 0 0 0 0 0
-0.8880 -0.1190 -2.2794 C 0 0 0 0 0 0 0 0 0 0 0 0
-2.2452 0.2814 -1.7680 C 0 0 0 0 0 0 0 0 0 0 0 0
-2.3767 0.0761 -0.3741 O 0 0 0 0 0 0 0 0 0 0 0 0
-3.5847 0.3988 0.2661 C 0 0 0 0 0 0 0 0 0 0 0 0
-3.7726 0.2011 1.7312 C 0 0 0 0 0 0 0 0 0 0 0 0
-2.5745 -0.3462 2.3474 N 0 0 0 0 0 0 0 0 0 0 0 0
-2.6755 -0.7662 3.4997 N 0 0 0 0 0 0 0 0 0 0 0 0
-2.7970 -1.1810 4.6516 N 0 0 0 0 0 0 0 0 0 0 0 0
-4.5686 0.8733 -0.3778 O 0 0 0 0 0 0 0 0 0 0 0 0
2.9042 -0.7628 1.7728 C 0 0 0 0 0 0 0 0 0 0 0 0
2.8269 0.5056 2.4707 N 0 0 0 0 0 0 0 0 0 0 0 0
3.8979 1.1083 2.6359 N 0 0 0 0 0 0 0 0 0 0 0 0
4.9600 1.7005 2.8196 N 0 0 0 0 0 0 0 0 0 0 0 0
2.3294 1.9098 -1.8628 H 0 0 0 0 0 0 0 0 0 0 0 0
3.4532 0.6158 -2.2512 H 0 0 0 0 0 0 0 0 0 0 0 0
1.3719 0.2700 -3.3808 H 0 0 0 0 0 0 0 0 0 0 0 0
1.4020 -0.9969 -2.1507 H 0 0 0 0 0 0 0 0 0 0 0 0
-0.8981 0.1227 -3.3791 H 0 0 0 0 0 0 0 0 0 0 0 0
-0.7538 -1.2195 -2.2505 H 0 0 0 0 0 0 0 0 0 0 0 0
-2.9654 -0.4393 -2.2357 H 0 0 0 0 0 0 0 0 0 0 0 0
-2.5204 1.2923 -2.0771 H 0 0 0 0 0 0 0 0 0 0 0 0
-4.5738 -0.5467 1.8719 H 0 0 0 0 0 0 0 0 0 0 0 0
-4.0527 1.1368 2.2571 H 0 0 0 0 0 0 0 0 0 0 0 0
3.7941 -1.3402 2.0839 H 0 0 0 0 0 0 0 0 0 0 0 0
2.0174 -1.3957 2.0696 H 0 0 0 0 0 0 0 0 0 0 0 0
1 2 2 0
2 3 1 0
3 4 1 0
4 5 1 0
5 6 1 0
6 7 1 0
7 8 1 0
8 9 1 0
9 10 1 0
10 11 1 0
11 12 1 0
12 13 2 0
13 14 2 0
10 15 2 0
2 16 1 0
16 17 1 0
17 18 2 0
18 19 2 0
4 20 1 0
4 21 1 0
5 22 1 0
5 23 1 0
7 24 1 0
7 25 1 0
8 26 1 0
8 27 1 0
11 28 1 0
11 29 1 0
16 30 1 0
16 31 1 0
M CHG 4 13 1 14 -1 18 1 19 -1
M END
76 changes: 76 additions & 0 deletions deepmd/deepmd_property_tools/DATA/mol_convert/id11.mol
Original file line number Diff line number Diff line change
@@ -0,0 +1,76 @@
id_11
RDKit 3D

34 35 0 0 0 0 0 0 0 0999 V2000
4.9473 2.1580 0.5415 O 0 0 0 0 0 0 0 0 0 0 0 0
4.4820 1.0513 0.1333 N 0 0 0 0 0 0 0 0 0 0 0 0
3.6345 1.0094 -1.0049 C 0 0 0 0 0 0 0 0 0 0 0 0
3.6970 1.9220 -1.9944 N 0 0 0 0 0 0 0 0 0 0 0 0
2.8050 1.6063 -2.8785 O 0 0 0 0 0 0 0 0 0 0 0 0
2.1563 0.5231 -2.5132 N 0 0 0 0 0 0 0 0 0 0 0 0
2.6533 0.1143 -1.3292 C 0 0 0 0 0 0 0 0 0 0 0 0
2.2114 -1.0051 -0.6059 N 0 0 0 0 0 0 0 0 0 0 0 0
2.4268 -2.2722 -1.2100 N 0 0 0 0 0 0 0 0 0 0 0 0
3.6878 -2.7819 -1.2793 O 0 0 0 0 0 0 0 0 0 0 0 0
1.4129 -2.8789 -1.6653 O 0 0 0 0 0 0 0 0 0 0 0 0
1.5690 -0.9931 0.6704 C 0 0 0 0 0 0 0 0 0 0 0 0
0.2847 -0.2866 0.6666 N 0 0 0 0 0 0 0 0 0 0 0 0
0.3222 1.0573 0.3261 N 0 0 0 0 0 0 0 0 0 0 0 0
1.1881 1.9636 0.8889 O 0 0 0 0 0 0 0 0 0 0 0 0
-0.4976 1.4313 -0.5653 O 0 0 0 0 0 0 0 0 0 0 0 0
-0.8778 -1.0603 0.9899 C 0 0 0 0 0 0 0 0 0 0 0 0
-2.0673 -0.2396 1.0549 N 0 0 0 0 0 0 0 0 0 0 0 0
-2.1773 0.6412 2.1189 N 0 0 0 0 0 0 0 0 0 0 0 0
-1.5044 1.8061 2.2984 O 0 0 0 0 0 0 0 0 0 0 0 0
-2.9962 0.3103 3.0304 O 0 0 0 0 0 0 0 0 0 0 0 0
-3.0316 -0.4124 0.0338 C 0 0 0 0 0 0 0 0 0 0 0 0
-2.9600 -1.3665 -0.9429 N 0 0 0 0 0 0 0 0 0 0 0 0
-4.0014 -1.2718 -1.7077 O 0 0 0 0 0 0 0 0 0 0 0 0
-4.7770 -0.3201 -1.3200 N 0 0 0 0 0 0 0 0 0 0 0 0
-4.2052 0.2489 -0.2263 C 0 0 0 0 0 0 0 0 0 0 0 0
-4.8032 1.3300 0.4498 N 0 0 0 0 0 0 0 0 0 0 0 0
-6.1981 1.3799 0.4700 O 0 0 0 0 0 0 0 0 0 0 0 0
-4.2250 2.2584 1.0198 O 0 0 0 0 0 0 0 0 0 0 0 0
4.8519 -0.0573 0.8533 O 0 0 0 0 0 0 0 0 0 0 0 0
1.4897 -1.9916 1.1005 H 0 0 0 0 0 0 0 0 0 0 0 0
2.1983 -0.3952 1.3741 H 0 0 0 0 0 0 0 0 0 0 0 0
-0.9612 -1.8291 0.1645 H 0 0 0 0 0 0 0 0 0 0 0 0
-0.7348 -1.6499 1.9262 H 0 0 0 0 0 0 0 0 0 0 0 0
1 2 2 0
2 3 1 0
3 4 2 0
4 5 1 0
5 6 1 0
6 7 2 0
7 8 1 0
8 9 1 0
9 10 1 0
9 11 2 0
8 12 1 0
12 13 1 0
13 14 1 0
14 15 1 0
14 16 2 0
13 17 1 0
17 18 1 0
18 19 1 0
19 20 1 0
19 21 2 0
18 22 1 0
22 23 2 0
23 24 1 0
24 25 1 0
25 26 2 0
26 27 1 0
27 28 1 0
27 29 2 0
2 30 1 0
7 3 1 0
26 22 1 0
12 31 1 0
12 32 1 0
17 33 1 0
17 34 1 0
M CHG 8 2 1 9 1 10 -1 14 1 15 -1 19 1 20 -1 27 1
M CHG 2 28 -1 30 -1
M END
Loading