38 lines
961 B
Clojure
Vendored
38 lines
961 B
Clojure
Vendored
(ns tech.v3.dataset.metamorph-test
|
|
(:require [tech.v3.dataset.metamorph :as ds-mm]
|
|
[tech.v3.dataset :as ds]
|
|
[clojure.test :as t :refer [deftest is]]))
|
|
|
|
(def df
|
|
(ds/->dataset "test/data/ames-train.csv.gz" {:key-fn keyword}))
|
|
|
|
|
|
(deftest call-with-df-1
|
|
(is (= [1 2 3 4 5]
|
|
(->>
|
|
((ds-mm/set-inference-target :SalePrice) df)
|
|
:metamorph/data
|
|
:Id
|
|
(take 5)
|
|
))))
|
|
|
|
(deftest call-with-df-2
|
|
(is (= [1 2 3 4 5]
|
|
(->>
|
|
((ds-mm/rename-columns {:SalePrice :sale-price :Id :id})
|
|
{:metamorph/data df})
|
|
((ds-mm/set-inference-target :sale-price) )
|
|
:metamorph/data
|
|
:id
|
|
(take 5)))))
|
|
|
|
(deftest brief
|
|
(let [df (ds/select-columns df (sort (ds/column-names df)))]
|
|
(is (= 334.0
|
|
(->>
|
|
((ds-mm/brief)
|
|
{:metamorph/data df})
|
|
:metamorph/data
|
|
first
|
|
:min)))))
|