From e78b1eec757f7098da832ac20b81a05a4c6c5fe6 Mon Sep 17 00:00:00 2001
From: Eric Hanson <5846501+ericphanson@users.noreply.github.com>
Date: Thu, 3 Oct 2024 16:26:09 +0200
Subject: [PATCH] fix tests: allow DataAPI as source for groupby (#87)

---
 test/runtests.jl | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/test/runtests.jl b/test/runtests.jl
index 1c915c6..c81dda6 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -97,7 +97,10 @@ if VERSION > v"1.9-"
         ext_imports = Dict(only_name_source(explicit_imports(TestPkg)))[DataFramesExt]
         @test ext_imports == [(; name=:DataFrames, source=DataFrames),
                               (; name=:DataFrame, source=DataFrames),
-                              (; name=:groupby, source=DataFrames)]
+                              (; name=:groupby, source=DataFrames)] ||
+              ext_imports == [(; name=:DataFrames, source=DataFrames),
+                              (; name=:DataFrame, source=DataFrames),
+                              (; name=:groupby, source=DataFrames.DataAPI)]
     end
 end