nera 0.0.1

data/test/test_nera_job_records.rb

@@ -0,0 +1,260 @@
+require File.dirname(__FILE__) + '/test_helper'
+
+class TC_NERA_JOB_RECS < Test::Unit::TestCase
+
+  def setup
+    @testdir = 'test_job_recs'
+    FileUtils.mkdir(@testdir)
+    FileUtils.chdir(@testdir)
+    @path_to_db = 'job_recs_table.marshal'
+    NERA::JobRecords.create_table( @path_to_db)
+    @db = NERA::JobRecords.new( @path_to_db)
+  end
+
+  def teardown
+    FileUtils.chdir('..')
+    FileUtils.rm_r(@testdir)
+  end
+
+  def test_constructor
+    assert_raise( RuntimeError) {
+      NERA::JobRecords.new('unknown_file')
+    }
+  end
+  
+  def test_create_table
+    f = NERA::JobRecords.create_table( @path_to_db)
+    assert_nil( f)
+  end
+
+  def test_keys
+    k = [ :id,:state,:simulator,:parameter_id,:run_id,
+          :number_of_runs,:created_at,:updated_at,:host_name ]
+    assert_equal( k , @db.keys)
+  end
+  
+  def test_add
+    assert_raise( ArgumentError) {
+      @db.add( 2,3,5,3)
+    }
+    assert_raise( ArgumentError) {
+      @db.add( "hoge",5,3,3)
+    }
+    assert_raise( ArgumentError) {
+      @db.add( 8,5,3,-1)
+    }
+
+    given_id = @db.add( Ising,5,3,4 )
+    d = DateTime.now
+    l = @db.list_all
+    rec = l[0]
+    assert_equal(1, given_id)
+    assert_equal( 1, rec[:id])
+    assert_equal( :created, rec[:state])
+    assert_equal( "Ising", rec[:simulator])
+    assert_equal( 5, rec[:parameter_id])
+    assert_equal( 3, rec[:run_id])
+    assert_equal( 4, rec[:number_of_runs] )
+    assert_equal( d.year, rec[:updated_at].year )
+    assert_equal( d.month, rec[:updated_at].month )
+    assert_equal( d.day, rec[:updated_at].day )
+    assert_equal( d.hour , rec[:updated_at].hour )
+    assert_equal( d.min, rec[:updated_at].min )
+    assert_equal( rec[:created_at], rec[:updated_at] )
+    assert_nil( rec[:host_name])
+    
+    sleep 2
+    given_id = @db.add( XY,4,2,5 )
+    d = DateTime.now
+    l = @db.list_all
+    rec = l[1]
+    assert_equal( 2, rec[:id])
+    assert_equal( :created, rec[:state])    
+    assert_equal( "XY", rec[:simulator])
+    assert_equal( 4, rec[:parameter_id])
+    assert_equal( 2, rec[:run_id])
+    assert_equal( 5, rec[:number_of_runs] )
+    assert_equal( d.year, rec[:updated_at].year )
+    assert_equal( d.month, rec[:updated_at].month )
+    assert_equal( d.day, rec[:updated_at].day )
+    assert_equal( d.hour , rec[:updated_at].hour )
+    assert_equal( d.min, rec[:updated_at].min )
+    assert_equal( rec[:created_at], rec[:updated_at] )
+    assert_nil( rec[:host_name])
+    
+    assert( l[0][:created_at] < l[1][:created_at])
+    
+  end
+
+  def add_three
+    @db.add( Ising,2,3,4)
+    @db.add( XY,6,7,8)
+    @db.add( Heisenberg,10,11,12)
+  end
+
+  def test_list
+    # ------
+    l = @db.list_all
+    assert( l.is_a?(Array) )
+    assert( l.size == 0 )
+
+    # ------
+    add_three
+    l = @db.list_all
+    assert( l.is_a?(Array) )
+    assert_equal( 3, l.size)
+    l2 = l.dup
+    l2.sort! do |a,b|
+      a[:id] <=> b[:id]
+    end
+    assert_equal( l, l2)
+    lc = @db.list_created
+    assert_equal( l, lc)
+  end
+
+  def test_find_by_id
+    assert_nil( @db.find_by_id(1) )
+    add_three
+    r = @db.find_by_id(1)
+    assert( r.is_a?(Hash) )
+    assert_equal( 1, r[:id])
+    assert_equal( :created, r[:state])    
+    assert_equal( "Ising", r[:simulator])
+    assert_equal( 2, r[:parameter_id])
+    assert_equal( 3, r[:run_id])
+    assert_equal( 4, r[:number_of_runs] )
+    assert_equal( r[:created_at], r[:updated_at] )
+    assert_nil( r[:host_name])
+
+    assert_nil( @db.find_by_id(500) )
+  end
+
+  def test_update_to_state_copied
+    add_three
+
+    r = @db.find_by_id(2)
+    assert_equal( :created, r[:state])
+    assert_nil( r[:host_name])
+
+    sleep 2
+    f = @db.update_to_state_copied( 2, "supacon.com")
+    assert_equal( true, f)
+    r2 = @db.find_by_id(2)
+    assert_equal( :copied, r2[:state])
+    assert_equal( r[:simulator], r2[:simulator])
+    assert_equal( r[:parameter_id], r2[:parameter_id])
+    assert_equal( r[:run_id], r2[:run_id])
+    assert_equal( r[:number_of_runs], r2[:number_of_runs])
+    assert_equal( r[:created_at], r2[:created_at])
+    assert( r[:updated_at] < r2[:updated_at])
+    assert_equal( "supacon.com", r2[:host_name])
+
+    la = @db.list_all
+    lc = @db.list_created
+    diff = la - lc
+    assert_equal( 1, diff.size)
+    assert_equal( r2, diff[0])
+
+    f = @db.update_to_state_copied( 2, "another_host")
+    assert_equal( true, f)
+    r2 = @db.find_by_id(2)
+    assert_equal( :copied, r2[:state])
+    assert_equal( "another_host", r2[:host_name])
+
+    assert_nil( @db.update_to_state_copied( 500, "unknown_host") )
+    assert_raise( ArgumentError) {
+      @db.update_to_state_copied( "hoge",2)
+    }
+    assert_raise( ArgumentError) {
+      @db.update_to_state_copied( 2, 2)
+    }
+    
+    lc2 = @db.list_created
+    assert_equal( lc, lc2)
+  end
+
+  def test_update_to_state_submitted
+    add_three
+
+    r = @db.find_by_id(2)
+    assert_equal( :created, r[:state])
+    assert_nil( r[:host_name])
+
+    sleep 2
+    f = @db.update_to_state_submitted( 2, "supacon.com")
+    assert_equal( true, f)
+    r2 = @db.find_by_id(2)
+    assert_equal( :submitted, r2[:state])
+    assert_equal( r[:simulator], r2[:simulator])
+    assert_equal( r[:parameter_id], r2[:parameter_id])
+    assert_equal( r[:run_id], r2[:run_id])
+    assert_equal( r[:number_of_runs], r2[:number_of_runs])
+    assert_equal( r[:created_at], r2[:created_at])
+    assert( r[:updated_at] < r2[:updated_at])
+    assert_equal( "supacon.com", r2[:host_name])
+
+    la = @db.list_all
+    lc = @db.list_created
+    diff = la - lc
+    assert_equal( 1, diff.size)
+    assert_equal( r2, diff[0])
+
+    f = @db.update_to_state_copied( 1, "another_host")
+    assert_equal( true, f)
+    f = @db.update_to_state_submitted( 1, "supacon.com")
+    r1 = @db.find_by_id(1)
+    assert_equal( :submitted, r1[:state])
+    assert_equal( "supacon.com", r1[:host_name])
+                  
+    assert_nil( @db.update_to_state_copied( 500, "unknown_host") )
+    assert_raise( ArgumentError) {
+      @db.update_to_state_copied( "hoge",2)
+    }
+    assert_raise( ArgumentError) {
+      @db.update_to_state_copied( 2, 2)
+    }
+    la = @db.list_all
+    lc = @db.list_created
+    diff = la - lc
+    assert_equal( 2, diff.size)
+  end
+
+  def test_destroy
+    add_three
+    r3 = @db.find_by_id(3)
+    destroyed = @db.destroy(3)
+    assert_equal( r3, destroyed)
+    l = @db.list_all
+    assert_equal( 2, l.size)
+
+    assert_nil( @db.destroy(3) )
+    @db.add( Ising,14,15,16)
+    l = @db.list_all
+    assert_equal( 3, l.size)
+    r4 = @db.find_by_id(4)
+    assert_equal(4, r4[:id])
+
+    assert_nil( @db.destroy(500) )
+
+    assert_raise(ArgumentError) {
+      @db.destroy("4")
+    }
+  end
+
+  def test_transaction
+    add_three
+    r3 = @db.find_by_id(3)
+    assert_raise(RuntimeError) {
+      @db.transaction {
+        destroyed = @db.destroy(3)
+        assert_equal( r3, destroyed)
+        l = @db.list_all
+        assert_equal( 2, l.size)
+        raise
+      }
+    }
+    assert_equal( r3, @db.find_by_id(3) )
+    assert_equal( 3, @db.list_all.size)
+  end
+
+end

data/test/test_nera_parameter_layer_controller.rb

@@ -0,0 +1,188 @@
+require File.dirname(__FILE__) + '/test_helper'
+
+
+class TC_NERA_PARAM_LAYER_CONTROLLER < Test::Unit::TestCase
+
+  def setup
+    @testdir = 'test_sim_controller'
+    FileUtils.mkdir(@testdir)
+    FileUtils.chdir(@testdir)
+    @db_folder = "nera_db"
+    NERA::DbFolders.create_db( @db_folder)
+    @dbf = NERA::DbFolders.new( @db_folder)
+    @slc = NERA::SimulatorLayerController.new( @db_folder)
+    @plc = NERA::ParameterLayerController.new( @db_folder, Ising)
+  end
+
+  def teardown
+    FileUtils.chdir('..')
+    FileUtils.rm_r(@testdir)
+  end
+
+  def test_constructor
+    assert_raise( ArgumentError) {
+      NERA::ParameterLayerController.new( @dbf,1)
+    }
+    assert_raise( ArgumentError) {
+      NERA::ParameterLayerController.new( Temp,1)
+    }
+    assert_raise( ArgumentError) {
+      x = NERA::ParameterLayerController.new( @dbf, "1")
+    }
+  end
+  
+  def test_parameters_list_in_csv
+    header, list = @plc.parameters_list_in_csv
+    assert( header.is_a?(String) )
+    assert_equal( "id, created_at, updated_at, L, K, tmax", header)
+    assert( list.is_a?(Array) )
+    assert_equal( 0, list.size)
+
+    f = @plc.create_a_new_parameter_set( {:L => 96, :K => 0.25, :tmax => 128} )
+    assert_equal( 1, f)
+    header, list = @plc.parameters_list_in_csv
+    assert( header.is_a?(String) )
+    assert_equal( "id, created_at, updated_at, L, K, tmax", header)
+    assert( list.is_a?(Array) )
+    assert_equal( 1, list.size)
+    dstr = DateTime.now.to_s.split('T')[0]
+    assert_equal( "1, #{dstr}, #{dstr}, 96, 0.25, 128", list[0])
+
+    f = @plc.create_a_new_parameter_set( {:L => 32} )
+    assert_equal( 2, f)
+    header, list = @plc.parameters_list_in_csv
+    assert_equal( 2, list.size)
+    assert_equal( "1, #{dstr}, #{dstr}, 96, 0.25, 128", list[0])
+    dstr = DateTime.now.to_s.split('T')[0]
+    assert_equal( "2, #{dstr}, #{dstr}, 32, 0.223, 512", list[1])
+
+    @plc.move_a_parameter_set_into_trashbox(1)
+    header, list = @plc.parameters_list_in_csv
+    assert_equal( 1, list.size)
+    assert_equal( "2, #{dstr}, #{dstr}, 32, 0.223, 512", list[0])
+  end
+
+  def add_three
+    f = @plc.create_a_new_parameter_set( {:L => 32, :K => 0.01, :tmax => 1} )
+    f = @plc.create_a_new_parameter_set( {:L => 64, :K => 0.02, :tmax => 2} )
+    f = @plc.create_a_new_parameter_set( {:L => 96, :K => 0.03, :tmax => 3} )
+  end
+
+  def test_get_id_from_csv_string
+    add_three
+    header, list = @plc.parameters_list_in_csv
+    assert_equal( 1, @plc.get_id_from_csv_string( list[0]) )
+    assert_equal( 2, @plc.get_id_from_csv_string( list[1]) )
+    assert_equal( 3, @plc.get_id_from_csv_string( list[2]) )
+  end
+  
+  def test_list_of_parameters
+    p1 = @plc.list_of_parameters
+    p2 = [ [:L, Integer, 32],
+           [:K, Float, 0.223],
+           [:tmax, Integer, 512]
+         ]
+    assert_equal( p2, p1)
+  end
+
+  def test_path_to_param_layer
+    assert_equal( "nera_db/Ising/", @plc.path_to_param_layer)
+  end
+
+  def test_create_a_new_parameter_set
+    f = @plc.create_a_new_parameter_set( {:L => 32, :K => 0.01, :tmax => 1} )
+    assert_equal( 1, f)
+    f = @plc.create_a_new_parameter_set( {:L => 64, :K => 0.02} )
+    assert_equal( 2, f)
+    f = @plc.create_a_new_parameter_set( {:L => 96, :tmax => 3} )
+    assert_equal( 3, f)
+    f = @plc.create_a_new_parameter_set( {:L => 64, :K => 0.02} )
+    assert_nil( f)
+    assert_raise( ArgumentError) {
+      f = @plc.create_a_new_parameter_set( {:L => 64, :unknown => 0.02} )
+    }
+    assert_raise( ArgumentError) {
+      f = @plc.create_a_new_parameter_set( {:L => 64, :K => 2} )
+    }
+    assert_raise( ArgumentError) {
+      f = @plc.create_a_new_parameter_set( [64,0.01,3] )
+    }
+    header, list = @plc.parameters_list_in_csv
+    d = DateTime.now.to_s.split('T')[0]
+    l = [ "1, #{d}, #{d}, 32, 0.01, 1",
+          "2, #{d}, #{d}, 64, 0.02, 512",
+          "3, #{d}, #{d}, 96, 0.223, 3"]
+    assert_equal( l, list)
+  end
+
+  def test_move_a_parameter_set_into_trashbox
+    add_three
+    f = @plc.move_a_parameter_set_into_trashbox(2)
+    assert_equal( true, f)
+    header, list = @plc.parameters_list_in_csv
+    assert_equal( 2, list.size)
+    f = @plc.move_a_parameter_set_into_trashbox(2)
+    assert_nil( f)
+    f = @plc.move_a_parameter_set_into_trashbox(5)
+    assert_nil( f)
+    assert_raise( ArgumentError) {
+      f = @plc.move_a_parameter_set_into_trashbox("1")
+    }
+  end
+
+  def test_trashbox_parameter_list_in_csv
+    add_three
+    h, l = @plc.parameters_list_in_csv
+    h2, l2 = @plc.trashbox_parameter_list_in_csv
+    assert_equal( h, h2)
+    assert_equal( [], l2)
+    @plc.move_a_parameter_set_into_trashbox(2)
+    h2, l2 = @plc.trashbox_parameter_list_in_csv
+    assert_equal( l[1], l2[0])
+  end
+  
+  def test_revert_a_parameter_set_in_trashbox
+    add_three
+    h_org, l_org = @plc.parameters_list_in_csv
+    f = @plc.move_a_parameter_set_into_trashbox(2)
+    f = @plc.revert_a_parameter_set_in_trashbox(2)
+    assert(f)
+    h2, l2 = @plc.parameters_list_in_csv
+    assert_equal( l_org, l2)
+    h3, l3 = @plc.trashbox_parameter_list_in_csv
+    assert_equal( 0, l3.size)
+
+    f = @plc.revert_a_parameter_set_in_trashbox(2)
+    assert_nil( f)
+    f = @plc.revert_a_parameter_set_in_trashbox(100)
+    assert_nil( f)
+
+    assert_raise( ArgumentError) {
+      @plc.revert_a_parameter_set_in_trashbox("2")
+    }
+  end
+
+  def test_delete_a_parameter_set
+    add_three
+    h_org, l_org = @plc.parameters_list_in_csv
+    @plc.move_a_parameter_set_into_trashbox(2)
+    s = @plc.delete_a_parameter_set(2)
+    assert(s)
+    ht, lt = @plc.trashbox_parameter_list_in_csv
+    assert_equal( [], lt)
+    
+    n = @plc.revert_a_parameter_set_in_trashbox(2)
+    assert_nil(n)
+    
+    n = @plc.revert_a_parameter_set_in_trashbox(100)
+    assert_nil(n)
+
+    n = @plc.revert_a_parameter_set_in_trashbox(1)
+    assert_nil(n)
+
+    assert_raise( ArgumentError) {
+      n = @plc.revert_a_parameter_set_in_trashbox("2")
+    }
+  end
+
+end