Use @test instead of @assert in test directory