Make <:AbstractOrbital behave like scalars under broadcasting (#97)

JuliaAtoms · Apr 26, 2021 · db0c019 · db0c019 · mortenpi · Apr 26, 2021
1 parent 2b3e59a
commit db0c019
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/orbitals.jl b/src/orbitals.jl
@@ -2,8 +2,13 @@
     abstract type AbstractOrbital
 
 Abstract supertype of all orbital types.
+
+!!! note "Broadcasting"
+
+    When broadcasting, orbital objects behave like scalars.
 """
 abstract type AbstractOrbital end
+Base.Broadcast.broadcastable(x::AbstractOrbital) = Ref(x)
 
 """
     const MQ = Union{Int,Symbol}

diff --git a/test/orbitals.jl b/test/orbitals.jl
@@ -424,4 +424,11 @@ using Random
             @test ne == e
         end
     end
+
+    @testset "Broadcasting" begin
+        @test ([o"1s", o"2p"] .== o"1s") == [true, false]
+        @test ([ro"1s", ro"2p-"] .== ro"1s") == [true, false]
+        @test ([so"1s(0,α)", so"2p(0,α)"] .== so"1s(0,α)") == [true, false]
+        @test ([rso"1s(1/2)", rso"2p-(1/2)"] .== rso"1s(1/2)") == [true, false]
+    end
 end