test: add failure mode tests

Author: avik-pal

Project.toml

@@ -34,10 +34,11 @@ julia = "1.10"
 
 [extras]
 Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
+EnzymeCore = "f151be2c-9106-41f4-ab19-57ee4f262869"
 ExplicitImports = "7d51a73a-1435-4ff3-83d9-f097790105c7"
 Optimisers = "3bd65402-5787-11e9-1adc-39752487f4e2"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
-test = ["Aqua", "ExplicitImports", "Optimisers", "Random", "Test"]
+test = ["Aqua", "EnzymeCore", "ExplicitImports", "Optimisers", "Random", "Test"]

test/runtests.jl

@@ -1,4 +1,4 @@
-using Aqua, ExplicitImports, Functors, LuxCore, Optimisers, Random, Test
+using Aqua, ExplicitImports, Functors, LuxCore, Optimisers, Random, Test, EnzymeCore
 
 rng = LuxCore._default_rng()
 
@@ -262,7 +262,7 @@ end
         @test check_no_self_qualified_accesses(LuxCore) === nothing
         @test check_all_explicit_imports_via_owners(LuxCore) === nothing
         @test check_all_qualified_accesses_via_owners(LuxCore) === nothing
-        @test check_all_explicit_imports_are_public(LuxCore) === nothing
+        @test_broken check_all_explicit_imports_are_public(LuxCore) === nothing
     end
 
     @testset "replicate" begin
@@ -279,4 +279,15 @@ end
         @test_broken length(fleaves(NamedTuple())) == 0  # upstream issue
         @test !LuxCore.check_fmap_condition(isodd, nothing, NamedTuple())
     end
+
+    @testset "Common Lux + Enzyme Mistakes" begin
+        d = Dense(2, 2)
+
+        @test_throws ArgumentError Active(d)
+        @test_throws ArgumentError Duplicated(d, d)
+        @test_throws ArgumentError DuplicatedNoNeed(d, d)
+        @test_throws ArgumentError BatchDuplicated(d, (d, d))
+        @test_throws ArgumentError BatchDuplicatedNoNeed(d, (d, d))
+        @test Const(d) isa Const
+    end
 end