change method locals to type object too (same as args)

test/typed/expressions/test_call.rb

@@ -26,7 +26,7 @@ module Register
     end
 
     def test_call_main_op
-      Register.machine.space.get_main.ensure_local(:bar , :Integer)
+      Register.machine.space.get_main.add_local(:bar , :Integer)
       @input    =s(:call,s(:name, :main),s(:arguments , s(:name, :bar) ))
       check
     end

test/typed/expressions/test_ops.rb

@@ -21,12 +21,12 @@ module Register
       end
     end
     def test_local_int
-      Register.machine.space.get_main.ensure_local(:bar , :Integer)
+      Register.machine.space.get_main.add_local(:bar , :Integer)
       @input    = s(:operator_value, :+, s(:name, :bar), s(:int, 3))
       check
     end
     def test_int_local
-      Register.machine.space.get_main.ensure_local(:bar , :Integer)
+      Register.machine.space.get_main.add_local(:bar , :Integer)
       @input    = s(:operator_value, :+, s(:int, 3), s(:name, :bar))
       check
     end

test/typed/expressions/test_vars.rb

@@ -10,7 +10,7 @@ module Register
     end
 
     def test_local
-      Register.machine.space.get_main.ensure_local(:bar , :Integer)
+      Register.machine.space.get_main.add_local(:bar , :Integer)
       @input    = s(:name, :bar)
       @output = Register::RegisterValue
       check

test/typed/parfait/test_method.rb

@@ -6,6 +6,8 @@ class TestMethod < MiniTest::Test
     obj = Register.machine.space.get_class_by_name(:Object)
     args = Parfait::Type.new_for_hash( obj , { bar: :Integer , foo: :Type})
     @method = ::Parfait::TypedMethod.new obj , :meth , args
+    @method.add_local :local_bar , :Integer
+    @method.add_local :local_foo , :Type
   end
 
   def test_method_name
@@ -39,7 +41,7 @@ class TestMethod < MiniTest::Test
     assert_equal 1 , index
     assert_equal :bar , @method.argument_name(index)
   end
-  def test_get_arg_name1
+  def test_get_arg_type1
     index = @method.has_arg(:bar)
     assert_equal :Integer , @method.argument_type(index)
   end
@@ -48,8 +50,46 @@ class TestMethod < MiniTest::Test
     assert_equal 2 , index
     assert_equal :foo , @method.argument_name(index)
   end
-  def test_get_arg_name2
+  def test_get_arg_type2
     index = @method.has_arg(:foo)
     assert_equal :Type , @method.argument_type(index)
   end
+
+  def test_local1
+    assert_equal 2 , @method.locals_length , @method.locals.inspect
+    assert_equal Symbol , @method.locals.first.class
+    assert_equal :local_bar , @method.locals_name(1)
+  end
+
+  def test_has_local
+    assert_equal 1 , @method.has_local(:local_bar)
+    assert_equal 2 , @method.has_local(:local_foo)
+  end
+
+  def test_add_local
+    @method.add_local(:foo2 , :Object)
+    assert_equal 3 , @method.locals_length
+    assert_equal :foo2 , @method.locals_name(3)
+    assert_equal :Object , @method.locals_type(3)
+  end
+
+  def test_get_locals_name1
+    index = @method.has_local(:local_bar)
+    assert_equal 1 , index
+    assert_equal :local_bar , @method.locals_name(index)
+  end
+  def test_get_locals_type1
+    index = @method.has_local(:local_bar)
+    assert_equal :Integer , @method.locals_type(index)
+  end
+  def test_get_locals_name2
+    index = @method.has_local(:local_foo)
+    assert_equal 2 , index
+    assert_equal :local_foo , @method.locals_name(index)
+  end
+  def test_get_locals_type2
+    index = @method.has_local(:local_bar)
+    assert_equal :Integer , @method.locals_type(index)
+  end
+
 end