using Test
using Sbplib
using Sbplib.Grids
using Sbplib.SbpOperators
using Sbplib.RegionIndices
using BenchmarkTools

import Sbplib.SbpOperators: Stencil


@testset "First" begin
    g = EquidistantGrid((10,10),(0.,0.), (1.,1.))
    v = evalOn(g, (x,y)->x^2+y^2+1)
    H = inner_product(g, 1., [1/2])

    @test  @ballocated(LazyTensors.apply($H.t1, $H.t2*$v, 1,2)) == 0
    @test  @ballocated(LazyTensors.apply($H.t1.tm, $(view(H.t2*v,:,1)), 2)) == 0

    # Nedan är halvdåliga
    @test  @ballocated(LazyTensors.apply($H.t1.tm, $(view(H.t2*v,1,:)), 2)) == 0
    @test  @ballocated(LazyTensors.apply($H.t1.tm, $(view(v,1,:)), 2)) == 0
    @test  @ballocated(LazyTensors.apply($H.t1.tm, $(view(v,:,1)), 2)) == 0
    @test  @ballocated(LazyTensors.apply($H.t1.tm, $(v[:,1]), 2)) == 0
end

@testset "Second" begin
    g = EquidistantGrid(10,0., 1.)
    v = evalOn(g, (x)->x^2+1)
    H = inner_product(g, 1., [1/2])
    V = SbpOperators.VolumeOperator(g, Stencil(1.,center=1), (Stencil(1/2,center=1),), SbpOperators.even)
    b = SbpOperators.BoundaryOperator(g, Stencil(1/2,center=1), Lower())

    @test @ballocated(LazyTensors.apply($H, $H*$v, 2)) == 0
    @test @ballocated(LazyTensors.apply($V, $V*$v, 2)) == 0
end