add import_module_only integration test and fix span hygiene

MusicalNinjaDad · Apr 18, 2024 · 4c1fced · 4c1fced
1 parent 4651717
commit 4c1fced
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/pyo3-testing/src/lib.rs b/pyo3-testing/src/lib.rs
@@ -139,11 +139,11 @@ fn wrap_testcase(testcase: Pyo3TestCase) -> TokenStream2 {
             py_AttributeErrormsgs
                 .push("Failed to get ".to_string() + &py_functionname + " function");
             py_functionidents.push(Ident::new(&py_functionname, Span::call_site()));
-            py_moduleswithfnsidents.push(Ident::new(&py_modulename, Span::mixed_site()));
+            py_moduleswithfnsidents.push(Ident::new(&py_modulename, Span::call_site()));
             py_functionnames.push(py_functionname);
         };
         py_ModuleNotFoundErrormsgs.push("Failed to import ".to_string() + &py_modulename);
-        py_moduleidents.push(Ident::new(&py_modulename, Span::mixed_site()));
+        py_moduleidents.push(Ident::new(&py_modulename, Span::call_site()));
         py_modulenames.push(py_modulename);
         o3_moduleidents.push(import.o3_moduleident);
     }

diff --git a/tests/test_pyo3test.rs b/tests/test_pyo3test.rs
@@ -47,3 +47,17 @@ fn test_pyo3test_simple_case() {
     let expected_result = 2_isize;
     assert_eq!(result, expected_result);
 }
+
+#[pyo3test]
+#[pyo3import(py_adders: import adders)]
+fn test_pyo3test_import_module_only() {
+    let result: isize = adders
+        .getattr("addone")
+        .unwrap()
+        .call1((1_isize,))
+        .unwrap()
+        .extract()
+        .unwrap();
+    let expected_result = 2_isize;
+    assert_eq!(result, expected_result);
+}