Return collocates as DataFrame
Change-Id: I6cde1b08782d9e7bf3bc6ad4d170cdb4d25b404f
diff --git a/test/runtests.jl b/test/runtests.jl
index bc29554..f1e0388 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -44,10 +44,10 @@
@testset "DerekoVecs.jl: collocation analysis" begin
if (!isnothing(wpd19.cdb))
- println(wpd19.vocab[30])
- coll = get_collocates(wpd19, "werden")
- @test coll[1].ldaf > 10
- @test coll[1].ldaf > coll[3].ldaf
+ df = get_collocates(wpd19, "werden")
+ @test df.collocate[1] == "kann"
+ @test df.ldaf[1] > 10
+ @test df.ldaf[1] > df.ldaf[3]
end
end