WIP: episodes model tests

Author: seb231

test/cic/model_test.clj

@@ -16,18 +16,25 @@
                     {:sex "2", :care-status "N1", :legal-status "V3", :uasc "True", :dob "1999", :ceased "2017-07-18", :id "124", :report-year "2017", :placement "U2", :report-date "2017-06-10"}
                     {:sex "2", :care-status "N1", :legal-status "V4", :uasc "True", :dob "1999", :ceased "2017-07-18", :id "124", :report-year "2017", :placement "U2", :report-date "2017-06-10"}))
 
-(def example (p/prepare-ages (->> example-data
-                                  (map c/format-episode)
-                                  c/episodes
-                                  c/episodes->periods)
-                             (r/make-random 50)))
+o(def most-recent {:sex "2", :care-status "E1", :legal-status "C2", :uasc "False", :dob "1999", :ceased "2017-07-18", :id "120", :report-year "2017", :placement "E2", :report-date "2017-04-18"})
+
+(defn prep-ages [data]
+  (p/prepare-ages (->> data
+                       (map c/format-episode)
+                       c/episodes
+                       c/episodes->periods)
+                  (r/make-random 50)))
+
+(def example-1 (prep-ages example-data))
+
+(def example-2 (prep-ages (conj example-data most-recent)))
 
 (deftest update-fuzzy-test
-  (let [age (:admission-age (first example))
+  (let [age (:admission-age (first example-1))
         fuzzy-age [(- age 1) age (+ age 1)]
-        yrs (/ (:duration (first example)) 365.0)
+        yrs (/ (:duration (first example-1)) 365.0)
         fuzzy-yrs (mapv int [(Math/floor yrs) (Math/ceil yrs)])
-        episodes (:episodes (first example))
+        episodes (:episodes (first example-1))
         result (update-fuzzy {} [age yrs] conj episodes)]
     (testing "returns range of ages close to example"
       (is (= (vec (distinct (map #(-> % first first) result)))
@@ -50,4 +57,4 @@
 
 (comment
   "build test around this"
-  ((episodes-model example) 0 (duration-model-0 (r/make-random 50)) (first example) (r/make-random 50)))
+  ((episodes-model example-2) 0 (duration-model-0 (r/make-random 50)) (nth example-1 3) (r/make-random 50)))