Add Tensorial scitype (#15)

* Add Tensorial scitype * Add more tests * Update CI.yml
JuliaML · Jun 5, 2024 · 7b93ac5 · 7b93ac5
1 parent 5f9fbde
commit 7b93ac5
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 1 deletion.
diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
@@ -20,9 +20,11 @@ jobs:
       matrix:
         version:
           - '1.9'
-          - 'nightly'
+          - '1'
         os:
           - ubuntu-latest
+          - macos-latest
+          - windows-latest
         arch:
           - x64
     steps:

diff --git a/src/DataScienceTraits.jl b/src/DataScienceTraits.jl
@@ -48,6 +48,13 @@ Scientific type of geometrical data (See Meshes.jl)
 """
 abstract type Geometrical <: SciType end
 
+"""
+    Tensorial
+
+Scientific type of tensorial data (e.g. Vector, Matrix)
+"""
+abstract type Tensorial <: SciType end
+
 """
     Temporal
 
@@ -114,6 +121,7 @@ scitype(::Type{<:Number}) = Continuous
 scitype(::Type{<:Integer}) = Categorical
 scitype(::Type{<:AbstractChar}) = Categorical
 scitype(::Type{<:AbstractString}) = Categorical
+scitype(::Type{<:AbstractArray}) = Tensorial
 scitype(::Type{<:TimeType}) = Temporal
 scitype(::Type{Union{T,Missing}}) where {T} = scitype(T)
 

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -30,6 +30,14 @@ const DST = DataScienceTraits
     @test scitype('a') <: DST.Categorical
     @test scitype("a") <: DST.Categorical
 
+    # Tensorial
+    @test scitype(Vector) <: DST.Tensorial
+    @test scitype(Matrix) <: DST.Tensorial
+    @test scitype(Array) <: DST.Tensorial
+    @test scitype(rand(2)) <: DST.Tensorial
+    @test scitype(rand(2, 2)) <: DST.Tensorial
+    @test scitype(rand(2, 2, 2)) <: DST.Tensorial
+
     # Unknown
     @test scitype(Nothing) <: DST.Unknown
     @test scitype(Union{}) <: DST.Unknown