commit 370dcb5750c42fa2191cf9d2ce8dbc6b2051c370
author Marc Kupietz <kupietz@ids-mannheim.de> Sat Jul 30 18:41:10 2022 +0200
committer Marc Kupietz <kupietz@ids-mannheim.de> Sat Jul 30 18:41:10 2022 +0200
tree 7ec3912816a1ffbc188f3e05421c981055a9b479
parent 2c404e5640bae556930c897d620ee21194012d4c

Allow passing max number of collocates

Change-Id: Ida073dde970549974adff899891ae1f4b968b83a

test/runtests.jl

diff --git a/test/runtests.jl b/test/runtests.jl
index f1e0388..66e3609 100644
--- a/test/runtests.jl
+++ b/test/runtests.jl
@@ -1,5 +1,6 @@
 using Artifacts
 using DerekoVecs
+using DataFrames
 using Test
 
 
@@ -48,6 +49,8 @@
             @test df.collocate[1] == "kann"
             @test df.ldaf[1] > 10
             @test df.ldaf[1] > df.ldaf[3]
+            @test nrow(get_collocates(wpd19, 3, 1)) == 1
+            @test nrow(get_collocates(wpd19, 3, 2)) == 2
         end
     end