restructure the test folders. add test_common.

Author: Han Wang

deepmd/pt_expt/common.py

@@ -31,5 +31,5 @@ def to_torch_array(array: Any) -> torch.Tensor | None:
     if array is None:
         return None
     if torch.is_tensor(array):
-        return array
+        return array.to(device=env.DEVICE)
     return torch.as_tensor(array, device=env.DEVICE)

source/tests/pt_expt/descriptor/__init__.py

@@ -0,0 +1 @@
+# SPDX-License-Identifier: LGPL-3.0-or-later

…urce/tests/pt_expt/model/test_se_e2_a.py …tests/pt_expt/descriptor/test_se_e2_a.pysource/tests/pt_expt/model/test_se_e2_a.py renamed to source/tests/pt_expt/descriptor/test_se_e2_a.py source/tests/pt_expt/model/test_se_e2_a.py renamed to source/tests/pt_expt/descriptor/test_se_e2_a.py

@@ -0,0 +1 @@
+# SPDX-License-Identifier: LGPL-3.0-or-later

source/tests/pt_expt/utils/__init__.py

@@ -0,0 +1,25 @@
+# SPDX-License-Identifier: LGPL-3.0-or-later
+import importlib
+
+import numpy as np
+
+from deepmd.pt_expt.common import (
+    to_torch_array,
+)
+from deepmd.pt_expt.utils import (
+    env,
+)
+
+torch = importlib.import_module("torch")
+
+
+def test_to_torch_array_moves_device() -> None:
+    arr = np.arange(6, dtype=np.float32).reshape(2, 3)
+    tensor = to_torch_array(arr)
+    assert torch.is_tensor(tensor)
+    assert tensor.device == env.DEVICE
+
+    input_tensor = torch.as_tensor(arr, device=torch.device("cpu"))
+    output_tensor = to_torch_array(input_tensor)
+    assert torch.is_tensor(output_tensor)
+    assert output_tensor.device == env.DEVICE