fix tests: allow DataAPI as source for groupby (#87)

ericphanson · Oct 3, 2024 · e78b1ee · e78b1ee
1 parent 2362f41
commit e78b1ee
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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