require 'helper'

CYCLONE_LOW_RES_FOLDER = 'test/cyclone_low_res'
class TestBasics < Test::Unit::TestCase
	def setup
    FileUtils.makedirs('test/slab_itg')
    @runner = CodeRunner.fetch_runner(Y: 'test/slab_itg', C: 'gs2', X: '/dev/null')
  end
  def teardown
    FileUtils.rm('test/slab_itg/.code_runner_script_defaults.rb')
    FileUtils.rm('test/slab_itg/.CODE_RUNNER_TEMP_RUN_LIST_CACHE')
    FileUtils.rmdir('test/slab_itg')
  end
  def test_basics
    assert_equal(@runner.run_class, CodeRunner::Gs2)
  end
end
if ENV['GS2_EXEC']
	class TestSubmission < Test::Unit::TestCase
		def setup
			CodeRunner.setup_run_class('gs2')
			CodeRunner::Gs2.make_new_defaults_file('test_gs2crmod', 'test/cyclone_low_res.in')
			FileUtils.mv('test_gs2crmod_defaults.rb', CodeRunner::Gs2.rcp.user_defaults_location + '/.')
			FileUtils.makedirs(tfolder)
		end
		def tfolder
			CYCLONE_LOW_RES_FOLDER
		end
		def teardown
			FileUtils.rm(CodeRunner::Gs2.rcp.user_defaults_location + '/' + 'test_gs2crmod_defaults.rb')
			FileUtils.rm(tfolder + '/.CODE_RUNNER_TEMP_RUN_LIST_CACHE')
			FileUtils.rm(tfolder + '/v/id_1/.code_runner_run_data')
			FileUtils.rm(tfolder + '/v/id_1/code_runner_results.rb')
			# Don't uncomment the line below unless you *really* know what you are doing! Replacing the test archive will break many of the tests
			Dir.chdir('test'){system "tar -czf cyclone_low_res.tgz cyclone_low_res/" unless FileTest.exist?('cyclone_low_res.tgz')}
			FileUtils.rm_r(tfolder)
		end
		def test_submission
			CodeRunner.submit(C: 'gs2', X: ENV['GS2_EXEC'], D: 'test_gs2crmod', n: '1', Y: tfolder, p: '{write_moments: ".true."}')
			CodeRunner.status(Y: tfolder)
		end
	end
else
	puts "\n************************************\nWarning: submission tests not run. Please specify the evironment variable GS2_EXEC (the path to the gs2 executable) if you wish to test submission.\n************************************\n"
	sleep 0.1
end
class TestAnalysis < Test::Unit::TestCase
	def setup
		Dir.chdir('test'){assert(system "tar -xzf cyclone_low_res.tgz")}
		@runner = CodeRunner.fetch_runner(Y: tfolder)
		@run = @runner.run_list[1]
	end
	def test_analysis
		assert_equal(1, @runner.run_list.size)
		assert_equal(0.13066732664774272, @runner.run_list[1].max_growth_rate)
		assert_equal(0.13066732664774272, @runner.run_list[1].growth_rate_at_ky[0.5])
		assert_equal(:Complete, @runner.run_list[1].status)
	end
	def test_interpolation
		assert_equal(5, @run.gsl_vector('kx').size)
		assert_equal(17, @run.gsl_vector('kx', interpolate_x: 4).size)
		kxvec = @run.gsl_vector('kx', interpolate_x: 4)
		assert_equal(0.0, kxvec[(kxvec.size-1)/2])
		xvec4 = @run.gsl_vector('x', interpolate_x: 4)
		s = kxvec.size
		assert_equal((-(s-1)/2..(s-1)/2).to_a.map{|i| i.to_f * kxvec.to_box_order[1]}.to_gslv, kxvec)
		#p xvec4.to_a
		xvec = @run.gsl_vector('x')
		# The vectors don't contain the periodic point, but we can construct them 
		# and they should be equal
		assert_equal(xvec[-1] + xvec[-1] - xvec[-2], xvec4[-1] + xvec4[-1] - xvec4[-2])
		#p  @run.gsl_vector('kx', interpolate_x: 4).to_a
		kyvec4 = @run.gsl_vector('ky', interpolate_y: 4)
		#p kyvec4
		s = kyvec4.size
		assert_equal((0...s).to_a.map{|i| i.to_f * kyvec4[1]}.to_gslv, kyvec4)
		yvec4 = @run.gsl_vector('y', interpolate_y: 4)
		#p yvec4.to_a
		yvec = @run.gsl_vector('y')
		assert_equal(yvec[-1] + yvec[-1] - yvec[-2], yvec4[-1] + yvec4[-1] - yvec4[-2])
		#kit = @runner.run_list[1].graphkit('phi_real_space_poloidal_plane', {n0: 1, Rgeo: 3, interpolate_theta: 8, torphi: Math::PI/4.0, interpolate_x: 8})
		kit1 = @runner.run_list[1].graphkit('phi_real_space_surface', {n0: 1, Rgeo: 3, gs2_coordinate_factor: 1.0})
		shape = kit1.data[0].x.data.shape
		shape = shape.map{|s| (s-1)*2 +1}
		kit = @runner.run_list[1].graphkit('phi_real_space_surface', {n0: 1, Rgeo: 3, interpolate_theta: 2, interpolate_x: 2, interpolate_y: 2, gs2_coordinate_factor: 1.0})
		assert_equal(shape, kit.data[0].x.data.shape)
		assert_equal(shape, kit.data[0].f.data.shape)
		#kit.gp.view = ["equal xyz", ",,5.0"]
		#kit.gnuplot 
	end
	def test_graphs
		kit = @runner.run_list[1].graphkit('phi2_by_ky_vs_time', {ky_index: 2})
		#kit.gnuplot
		assert_equal(51, kit.data[0].y.data.size)
		assert_equal(@runner.run_list[1].netcdf_file.var('phi2_by_ky').get('start' => [1,4], 'end' => [1,4]).to_a[0][0], kit.data[0].y.data[4])
        
        kit = @run.graphkit('tpar2_by_mode_vs_time', {ky_index:2, kx_index:1, species_index:1})
        #kit.gnuplot
		assert_equal(@runner.run_list[1].netcdf_file.var('tpar2_by_mode').get('start' => [0,1,0,4], 'end' => [0,1,0,4]).to_a[0][0][0][0], kit.data[0].y.data[4])

        kit = @run.graphkit('tperp2_by_mode_vs_time', {ky_index:2, kx_index:1, species_index:1})
        #kit.gnuplot
		assert_equal(@runner.run_list[1].netcdf_file.var('tperp2_by_mode').get('start' => [0,1,0,4], 'end' => [0,1,0,4]).to_a[0][0][0][0], kit.data[0].y.data[4])
	end
	def test_3d_graphs
		kit = @runner.run_list[1].graphkit('phi_real_space', {n0: 3, Rgeo: 3})
		assert_equal([5,5,9], kit.data[0].f.data.shape)
		assert_equal(-0.00402, kit.data[0].f.data[2,3,6].round(5))
		kit = @runner.run_list[1].graphkit('density_real_space', {n0: 3, Rgeo: 3, species_index: 1, gs2_coordinate_factor: 0.9})
		#kit.gnuplot
		assert_equal([5,5,9], kit.data[0].f.data.shape)
		assert_equal(-0.00985, kit.data[0].f.data[2,3,6].round(5))
		kit = @runner.run_list[1].graphkit('phi_real_space', {n0: 3, Rgeo: 3, t_index: 2})
		assert_equal([5,5,9], kit.data[0].f.data.shape)
		assert_equal(0.00031, kit.data[0].f.data[2,3,6].round(5))
		kit = @runner.run_list[1].graphkit('phi_real_space_surface', {n0: 3, Rgeo: 3, interpolate_theta: 2})
		assert_equal([5,5,1], kit.data[0].f.data.shape)
		assert_equal([5,1,17], kit.data[2].f.data.shape)
		assert_equal(-0.00153, kit.data[0].f.data[1,4,0].round(5))
		#kit.gnuplot
		kit = @runner.run_list[1].graphkit('phi_real_space_poloidal_plane', {n0: 1, Rgeo: 3, interpolate_theta: 8, torphi: Math::PI/4.0})
		assert_equal(-0.00208, kit.data[0].f.data[-1,1].round(5))
		assert_equal(1.707, kit.data[0].x.data[-1,1].round(3))
		kit.gp.view = ["equal xyz", ",,4.0"]
		#kit.gnuplot
		kit = @runner.run_list[1].graphkit('density_real_space_poloidal_plane', {n0: 1, Rgeo: 3, interpolate_theta: 8, torphi: Math::PI/4.0, species_index: 1})
		assert_equal(-0.00352, kit.data[0].f.data[-1,1].round(5))
		assert_equal(1.707, kit.data[0].x.data[-1,1].round(3))
		kit.gp.view = ["equal xyz", ",,4.0"]

		kit = @runner.run_list[1].graphkit('density_real_space_poloidal_plane', {n0: 1, Rgeo: 3, interpolate_theta: 8, torphi: Math::PI/4.0, species_index: 1, t_index: 50})
		assert_equal(-0.00208, kit.data[0].f.data[-1,1].round(5))
		assert_equal(1.707, kit.data[0].x.data[-1,1].round(3))
		kit.gp.view = ["equal xyz", ",,4.0"]
		#kit.gnuplot
		kit = @runner.run_list[1].graphkit('phi_real_space_standard_representation', {n0: 1, Rgeo: 3, interpolate_theta: 2, torphi_values: [Math::PI/4.0,3.0*Math::PI/4.0], interpolate_y: 2})
		assert_equal([5,17], kit.data[0].f.data.shape)
		assert_equal([3,1,17], kit.data[2].f.data.shape)
		assert_equal([3,1,17], kit.data[2].y.data.shape)
		assert_equal(-0.00038, kit.data[0].f.data[-1,1].round(5))
		assert_equal(2.12132, kit.data[2].y.data[2,0,12].round(5))
		#kit.gnuplot 
	end

	def tfolder
		CYCLONE_LOW_RES_FOLDER
	end
	def teardown
		FileUtils.rm_rf(tfolder)
	end
end

AGK_SLAB_ITG_LOW_KPERP_FOLDER = 'test/agk_slab_itg_low_kperp'

class TestAgkAnalysis < Test::Unit::TestCase

	def setup
		Dir.chdir('test'){assert(system "tar -xzf agk_slab_itg_low_kperp.tgz")}
		@runner = CodeRunner.fetch_runner(Y: tfolder)
		@run = @runner.run_list[1]
	end
	def tfolder
		AGK_SLAB_ITG_LOW_KPERP_FOLDER
	end
	def test_graphs
		kit = @runner.run_list[1].graphkit('phi2_by_ky_vs_time', {ky_index: 2})
		#kit.gnuplot
		assert_equal(126, kit.data[0].y.data.size)
		assert_equal(@runner.run_list[1].netcdf_file.var('phi2').get('start' => [4], 'end' => [4]).to_a[0], kit.data[0].y.data[4])
    assert_equal(nil,@runner.run_list[1].instance_variable_get(:@shat))
		kit = @runner.run_list[1].graphkit('kpar_spectrum',{})
		#kit.gnuplot
	end
	def test_analysis
		assert_equal(1, @runner.run_list.size)
		assert_equal(0.03224, @runner.run_list[1].max_growth_rate.round(5))
		#p @runner.run_list[1].growth_rate_at_ky
		assert_equal(0.03224, @runner.run_list[1].growth_rate_at_ky[0.01].round(5))
		assert_equal(:Complete, @runner.run_list[1].status)
	end
	def teardown
		FileUtils.rm_rf(tfolder)
	end
end



if ENV['AGK_EXEC']
	class TestAstrogkSubmission < Test::Unit::TestCase
		def setup
			CodeRunner.setup_run_class('gs2', modlet: 'astrogk')
			CodeRunner::Gs2::Astrogk.make_new_defaults_file('test_gs2crmod_astrogk', 'test/agk_slab_itg_low_kperp.in')
			FileUtils.mv('test_gs2crmod_astrogk_defaults.rb', CodeRunner::Gs2::Astrogk.rcp.user_defaults_location + '/.')
			FileUtils.makedirs(tfolder)
		end
		def tfolder
			AGK_SLAB_ITG_LOW_KPERP_FOLDER
		end
		def teardown
			FileUtils.rm(CodeRunner::Gs2::Astrogk.rcp.user_defaults_location + '/' + 'test_gs2crmod_astrogk_defaults.rb')
			FileUtils.rm(tfolder + '/.CODE_RUNNER_TEMP_RUN_LIST_CACHE')
			FileUtils.rm(tfolder + '/v/id_1/.code_runner_run_data')
			FileUtils.rm(tfolder + '/v/id_1/code_runner_results.rb')
			 #Don't uncomment the line below unless you *really* know what you are doing! Replacing the test archive will break many of the tests
			Dir.chdir('test'){system "tar -czf agk_slab_itg_low_kperp.tgz agk_slab_itg_low_kperp/" unless FileTest.exist?('agk_slab_itg_low_kperp.tgz')}
			FileUtils.rm_r(tfolder)
		end
		def test_submission
			CodeRunner.submit(C: 'gs2', X: ENV['AGK_EXEC'], D: 'test_gs2crmod_astrogk', n: '4', Y: tfolder, m: 'astrogk')
			
			CodeRunner.status(Y: tfolder)
		end
	end
else
	puts "\n************************************\nWarning: agk submission tests not run. Please specify the evironment variable AGK_EXEC (the path to the agk executable) if you wish to test submission.\n************************************\n"
	sleep 0.1
end

class TestBasicsSpectroGK < Test::Unit::TestCase
	def setup
    @runner = CodeRunner.fetch_runner(Y: 'test/slab_itg', C: 'gs2', X: '/dev/null', m: 'spectrogk')
  end
  def teardown
    FileUtils.rm('test/slab_itg/.code_runner_script_defaults.rb')
    #FileUtils.rm('test/slab_itg/.CODE_RUNNER_TEMP_RUN_LIST_CACHE')
  end
  def test_basics
    assert_equal(@runner.run_class, CodeRunner::Gs2::Spectrogk)
  end
	def test_variables
		assert_equal(Hash, @runner.run_class.rcp.namelists.class)
		assert_equal(@runner.run_class.rcp.namelists[:layouts_knobs].class, Hash)
		assert_equal(@runner.run_class.rcp.namelists[:parameters_knobs].class, Hash)
		assert_equal(@runner.run_class.rcp.namelists[:parameters_knobs][:variables][:force_5d].class, Hash)
	end
end