adds tests for dataset label helper fns
This commit is contained in:
parent
6a30d86841
commit
99048715f1
2 changed files with 8 additions and 2 deletions
|
@ -184,8 +184,8 @@
|
|||
|
||||
(defn dataset-class-labels
|
||||
"Returns the possible labels for the class attribute"
|
||||
[dataset]
|
||||
(let [class-attr (.classAttribute dataset)
|
||||
[^Instances dataset]
|
||||
(let [^Attribute class-attr (.classAttribute dataset)
|
||||
values (.enumerateValues class-attr)]
|
||||
(loop [continue (.hasMoreElements values)
|
||||
acum {}]
|
||||
|
|
|
@ -146,3 +146,9 @@
|
|||
(let [ds (make-dataset "test" [:a {:b [:foo :bar]}] [[1 :foo] [2 :bar]])
|
||||
_ (dataset-replace-attribute! ds :b (nominal-attribute :b [:baz :shaz]))]
|
||||
(is (= [:a {:b [:shaz :baz]}] (dataset-format ds)))))
|
||||
|
||||
(deftest dataset-label-helpers
|
||||
(let [ds (make-dataset "test" [:a :b {:c [:d :e]}]
|
||||
[{:a 1 :b 2 :c :d} [4 5 :e]])]
|
||||
(dataset-set-class ds :c)
|
||||
(is (= {:d 0 :e 1} (dataset-class-labels ds) (dataset-labels-at ds 2)))))
|
||||
|
|
Loading…
Reference in a new issue