put fold! method in separate file

phrb · Jul 15, 2021 · 54324c8 · 54324c8
1 parent cb6a26c
commit 54324c8
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 34 deletions.
diff --git a/src/ExperimentalDesign.jl b/src/ExperimentalDesign.jl
@@ -35,5 +35,6 @@ include("random.jl")
 include("kl_exchange.jl")
 include("custom_show.jl")
 include("categorical.jl")
+include("fold.jl")
 
 end
diff --git a/src/design.jl b/src/design.jl
@@ -97,40 +97,6 @@ end
 """
 $(TYPEDSIGNATURES)
 
-```jldoctest
-julia> fold!(PlackettBurman(2))
-8×3 DataFrame
-│ Row │ factor1 │ factor2 │ dummy1 │
-│     │ Int64   │ Int64   │ Int64  │
-├─────┼─────────┼─────────┼────────┤
-│ 1   │ 1       │ 1       │ 1      │
-│ 2   │ 1       │ -1      │ -1     │
-│ 3   │ -1      │ -1      │ 1      │
-│ 4   │ -1      │ 1       │ -1     │
-│ 5   │ -1      │ -1      │ -1     │
-│ 6   │ -1      │ 1       │ 1      │
-│ 7   │ 1       │ 1       │ -1     │
-│ 8   │ 1       │ -1      │ 1      │
-
-```
-"""
-function fold!(design::AbstractScreeningDesign)
-    mirror_matrix = deepcopy(design.matrix)
-    if mirror_matrix[1,1] isa Symbol
-        for col in names(mirror_matrix)
-            replace!(mirror_matrix[col], :low => :high, :high => :low)
-        end
-    else
-        for col in names(mirror_matrix)
-            replace!(mirror_matrix[col], 1 => -1, -1 => 1)
-        end
-    end
-    append!(design.matrix, mirror_matrix)
-end
-
-"""
-$(TYPEDSIGNATURES)
-
 ```jldoctest
 julia> PlackettBurman(4)
 PlackettBurman

diff --git a/src/fold.jl b/src/fold.jl
@@ -0,0 +1,33 @@
+"""
+$(TYPEDSIGNATURES)
+
+```jldoctest
+julia> fold!(PlackettBurman(2))
+8×3 DataFrame
+│ Row │ factor1 │ factor2 │ dummy1 │
+│     │ Int64   │ Int64   │ Int64  │
+├─────┼─────────┼─────────┼────────┤
+│ 1   │ 1       │ 1       │ 1      │
+│ 2   │ 1       │ -1      │ -1     │
+│ 3   │ -1      │ -1      │ 1      │
+│ 4   │ -1      │ 1       │ -1     │
+│ 5   │ -1      │ -1      │ -1     │
+│ 6   │ -1      │ 1       │ 1      │
+│ 7   │ 1       │ 1       │ -1     │
+│ 8   │ 1       │ -1      │ 1      │
+
+```
+"""
+function fold!(design::AbstractScreeningDesign)
+    mirror_matrix = deepcopy(design.matrix)
+    if mirror_matrix[1,1] isa Symbol
+        for col in names(mirror_matrix)
+            replace!(mirror_matrix[col], :low => :high, :high => :low)
+        end
+    else
+        for col in names(mirror_matrix)
+            replace!(mirror_matrix[col], 1 => -1, -1 => 1)
+        end
+    end
+    append!(design.matrix, mirror_matrix)
+end