add tests for construction with abstract vectors

m3g · Sep 14, 2023 · 3c82e7f · 3c82e7f
1 parent de9771b
commit 3c82e7f
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/src/PeriodicSystems.jl b/src/PeriodicSystems.jl
@@ -831,6 +831,23 @@ end
     sys = PeriodicSystem(xpositions=x, ypositions=y, unitcell=[1.0, 1.0, 1.0], cutoff=0.1, output=0.0, parallel=false)
     a = @ballocated PeriodicSystems.UpdatePeriodicSystem!($sys) samples = 1 evals = 1
     @test a == 0
+
+    # Test construction with more general abstract vectors
+    x = @view(x[1:500])
+    sys = PeriodicSystem(xpositions=x, unitcell=[1.0, 1.0, 1.0], cutoff=0.1, output=0.0, parallel=false)
+    a = @ballocated PeriodicSystems.UpdatePeriodicSystem!($sys) samples = 1 evals = 1
+    @test a == 0
+    y = @view(y[1:500])
+    sys = PeriodicSystem(xpositions=x, ypositions=y, unitcell=[1.0, 1.0, 1.0], cutoff=0.1, output=0.0, parallel=false)
+    a = @ballocated PeriodicSystems.UpdatePeriodicSystem!($sys) samples = 1 evals = 1
+    @test a == 0
+
+    x = rand(3,500)
+    xr = reinterpret(reshape, SVector{3,Float64}, x)
+    sys = PeriodicSystem(xpositions=xr, unitcell=[1.0, 1.0, 1.0], cutoff=0.1, output=0.0, parallel=false)
+    a = @ballocated PeriodicSystems.UpdatePeriodicSystem!($sys) samples = 1 evals = 1
+    @test a == 0
+
 end
 
 """