Test split_string

MilesCranmer · Jun 19, 2023 · 9f78093 · 9f78093
1 parent 8ef0a46
commit 9f78093
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/test/test_print.jl b/test/test_print.jl
@@ -1,5 +1,6 @@
 using Test
 using SymbolicRegression
+using SymbolicRegression.UtilsModule: split_string
 
 include("test_params.jl")
 
@@ -45,3 +46,12 @@ for binop in [safe_pow, ^]
     minitree = Node(5, Node("x1"), Node("x2"))
     @test string_tree(minitree, opts) == "(x1 ^ x2)"
 end
+
+@testset "Test splitting of strings" begin
+    split_string("abcdefgh", 3) == ["abc", "def", "gh"]
+    split_string("abcdefgh", 100) == ["abcdefgh"]
+    split_string("⋅", 1) == ["⋅"]
+    split_string("⋅⋅", 1) == ["⋅", "⋅"]
+    split_string("⋅⋅⋅⋅", 2) == ["⋅⋅", "⋅⋅"]
+    split_string("ραβγ", 2) == ["ρα", "βγ"]
+end