1 files changed, 12 insertions, 5 deletions
diff --git a/rust/tests/base_detection.rs b/rust/tests/base_detection.rs
index b5497c3c..e4594b8e 100644
--- a/rust/tests/base_detection.rs
+++ b/rust/tests/base_detection.rs
@@ -1,4 +1,4 @@
-use binaryninja::base_detection::BaseAddressDetectionSettings;
+use binaryninja::base_detection::{BaseAddressDetectionConfidence, BaseAddressDetectionSettings};
 use binaryninja::binary_view::BinaryViewExt;
 use binaryninja::headless::Session;
 use rstest::{fixture, rstest};
@@ -11,14 +11,21 @@ fn session() -> Session {
 }
 
 #[rstest]
-fn test_failed_base_detection(_session: &Session) {
+fn test_base_detection(_session: &Session) {
     let out_dir = env!("OUT_DIR").parse::<PathBuf>().unwrap();
-    let view = binaryninja::load(out_dir.join("atox.obj")).expect("Failed to create view");
+    let view = binaryninja::load(out_dir.join("raw_base_detection_aarch64"))
+        .expect("Failed to create view");
     let bad = view
         .base_address_detection()
         .expect("Failed to create base address detection");
     assert!(
-        !bad.detect(&BaseAddressDetectionSettings::default()),
-        "Detection should fail on this view"
+        bad.detect(&BaseAddressDetectionSettings::default()),
+        "Detection should succeed on this view"
+    );
+    let result = bad.scores(10);
+    assert_eq!(result.scores.len(), 3);
+    assert_eq!(
+        result.confidence,
+        BaseAddressDetectionConfidence::HighConfidence
     );
 }