move test cases to "basic" testset
This commit is contained in:
parent
47728b1899
commit
c4ebd199db
@ -1,6 +1,7 @@
|
|||||||
using Test, Random
|
using Test, Random
|
||||||
import Flux: activations
|
import Flux: activations
|
||||||
|
|
||||||
|
@testset "basic" begin
|
||||||
@testset "helpers" begin
|
@testset "helpers" begin
|
||||||
@testset "activations" begin
|
@testset "activations" begin
|
||||||
dummy_model = Chain(Dense(10,5,σ),Dense(5,2),softmax)
|
dummy_model = Chain(Dense(10,5,σ),Dense(5,2),softmax)
|
||||||
@ -15,7 +16,6 @@ import Flux: activations
|
|||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
@testset "basic" begin
|
|
||||||
@testset "Chain" begin
|
@testset "Chain" begin
|
||||||
@test_nowarn Chain(Dense(10, 5, σ), Dense(5, 2))(randn(10))
|
@test_nowarn Chain(Dense(10, 5, σ), Dense(5, 2))(randn(10))
|
||||||
@test_throws DimensionMismatch Chain(Dense(10, 5, σ),Dense(2, 1))(randn(10))
|
@test_throws DimensionMismatch Chain(Dense(10, 5, σ),Dense(2, 1))(randn(10))
|
||||||
|
Loading…
Reference in New Issue
Block a user