init research
This commit is contained in:
@@ -0,0 +1,29 @@
|
||||
(ns tech.v3.dataset.object-columns-test
|
||||
(:require [clojure.test :refer [deftest is]]
|
||||
[tech.v3.dataset :as ds]
|
||||
[tech.v3.datatype :as dtype]
|
||||
[tech.v3.datatype.datetime :as dtype-dt]
|
||||
[tech.v3.tensor :as dtt]))
|
||||
|
||||
|
||||
(deftest basic-object-columns
|
||||
(let [src-ds (ds/->dataset {:a (range 10)
|
||||
:b (repeat 10 {:a 1 :b 2})})]
|
||||
(is (= :persistent-map
|
||||
(dtype/get-datatype (src-ds :b))))
|
||||
(is (= (vec (repeat 10 {:a 1 :b 2}))
|
||||
(vec (dtype/->reader (src-ds :b)))))))
|
||||
|
||||
|
||||
(deftest involved-object-columns
|
||||
(let [src-ds (ds/->dataset
|
||||
{:dates (list "2000-01-01" "2000-02-01" "2000-03-01"
|
||||
"2000-04-01" "2000-05-01")
|
||||
:integers (range 5)
|
||||
:durations (repeat 5 (dtype-dt/duration))
|
||||
:doubles (map double (range 5))
|
||||
:tensors (repeat 5 (dtt/->tensor (partition 2 (range 4))))})]
|
||||
(is (= #{:float64 :string :int64 :tensor
|
||||
:packed-duration}
|
||||
(->> (map dtype/get-datatype (vals src-ds))
|
||||
set)))))
|
||||
Reference in New Issue
Block a user