Contents

require File.join(File.expand_path(File.dirname(__FILE__)), '../..', 'test_helper.rb')
require 'rbbt/util/python'

class TestPython < Test::Unit::TestCase

  def test_python
    TmpFile.with_file do |tmpdir|
      code =<<-EOF
def python_test(a, b):
	c = a + b
	return c
      EOF
      Open.write(File.join(tmpdir, 'file1.py'), code)
      Open.write(File.join(tmpdir, 'file2.py'), code)
      Open.write(File.join(tmpdir, 'file3.py'), code)
      Open.write(File.join(tmpdir, 'file4.py'), code)
      RbbtPython.add_path tmpdir

      res = nil

      RbbtPython.run 'file2', :python_test do 
        res = python_test(1, 3)
      end
      assert_equal 4, res

      RbbtPython.run do
        pyfrom :file3, :import => :python_test
        res = python_test(1, 4)
      end
      assert_equal 5, res

      RbbtPython.run do
        pyimport :file4
        res = file4.python_test(1, 4)
      end
      assert_equal 5, res

      RbbtPython.run 'file1' do 
        res = file1.python_test(1, 2)
      end
      assert_equal 3, res
    end
  end

  def test_run_log
    TmpFile.with_file do |tmpdir|
      code =<<-EOF
import sys
def python_print():
	print("Test STDERR", file=sys.stderr)
	print("Test STDOUT")
      EOF
      Open.write(File.join(tmpdir, 'file_print.py'), code)
      RbbtPython.add_path tmpdir

      RbbtPython.run_log 'file_print' do 
        file_print.python_print
      end
      RbbtPython.run_log_stderr 'file_print' do 
        file_print.python_print
      end

      RbbtPython.run_log 'file_print' do 
        file_print.python_print
      end
    end
  end

  def test_keras
    defined = RbbtPython.run do
      pyimport "tensorflow.keras.models", as: :km
      defined?(km.Sequential)
    end
    assert defined
  end

  def test_keras_import
    defined = RbbtPython.run do
      pyfrom "tensorflow.keras.models", import: :Sequential
      defined?(self::Sequential)
    end
    assert defined
  end
end

Version data entries

63 entries across 63 versions & 1 rubygems

Version	Path
rbbt-util-5.35.1	test/rbbt/util/test_python.rb
rbbt-util-5.34.27	test/rbbt/util/test_python.rb
rbbt-util-5.34.26	test/rbbt/util/test_python.rb
rbbt-util-5.34.25	test/rbbt/util/test_python.rb
rbbt-util-5.34.24	test/rbbt/util/test_python.rb
rbbt-util-5.34.23	test/rbbt/util/test_python.rb
rbbt-util-5.34.22	test/rbbt/util/test_python.rb
rbbt-util-5.34.21	test/rbbt/util/test_python.rb
rbbt-util-5.34.20	test/rbbt/util/test_python.rb
rbbt-util-5.34.18	test/rbbt/util/test_python.rb
rbbt-util-5.34.17	test/rbbt/util/test_python.rb
rbbt-util-5.34.16	test/rbbt/util/test_python.rb
rbbt-util-5.34.15	test/rbbt/util/test_python.rb
rbbt-util-5.34.14	test/rbbt/util/test_python.rb
rbbt-util-5.34.13	test/rbbt/util/test_python.rb
rbbt-util-5.34.12	test/rbbt/util/test_python.rb
rbbt-util-5.34.11	test/rbbt/util/test_python.rb
rbbt-util-5.34.10	test/rbbt/util/test_python.rb
rbbt-util-5.34.9	test/rbbt/util/test_python.rb
rbbt-util-5.34.8	test/rbbt/util/test_python.rb