From 4342988027ee0811b57d3641bda4c8486d7da1f5 Mon Sep 17 00:00:00 2001
From: Zach Burnett <zburnett@stsci.edu>
Date: Mon, 6 May 2024 13:44:21 -0400
Subject: [PATCH] ensure extension modules are present in built wheel (#8462)

---
 .github/workflows/build.yml | 1 +
 setup.py                    | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
index ca5fb0d53e..f94367e4b1 100644
--- a/.github/workflows/build.yml
+++ b/.github/workflows/build.yml
@@ -32,5 +32,6 @@ jobs:
         # Until we have arm64 runners, we can't automatically test arm64 wheels
         - cp3*-macosx_arm64
       sdist: true
+      test_command: python -c "from jwst.lib import winclip; from jwst.cube_build import cube_match_internal, cube_match_sky_pointcloud, cube_match_sky_driz, blot_median; from jwst.straylight import calc_xart"
     secrets:
       pypi_token: ${{ secrets.PYPI_PASSWORD_STSCI_MAINTAINER }}
diff --git a/setup.py b/setup.py
index 245ed5f6aa..d99ad0adcc 100644
--- a/setup.py
+++ b/setup.py
@@ -8,6 +8,8 @@
 define_macros = [("NUMPY", "1")]
 
 setup(
+    # importing these extension modules is tested in `.github/workflows/build.yml`; 
+    # when adding new modules here, make sure to add them to the `test_command` entry there
     ext_modules=[
         Extension(
             "jwst.lib.winclip",