(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)))))