$title 'Test for embedded code facility' (EMBPY01,SEQ=735)

$onText
Test functionality of the GAMS embedded code facility for Python during compilation time

Contributor: Clemens Westphal, July 2017
$offText


$log --- Using Python library %sysEnv.GMSPYTHONLIB%


Set i       / i1 text for i1
              i2 text for i2 /
    iout(i)
    j       / j1*j3 /
    ij(i,j) / #i.#j /
    i_want(i) / i1, i2 /
    i_want_ex(i) / i1 text 1, i2 text 2 /
    ij_want(i,j) / i1.j1, i1.j2 /
    ij_want_merge(i,j) / i1.j1, i1.j2, i1.j3 /
    ij_want_ex(i,j) / i1.j1 text 1, i1.j2 text 2/;

Scalar p0 /3.14/
       p0_want / 3.14 /;

Parameter p1(i)   / #i 3.14 /
          p2(i,j) / #i.#j 3.14 /
          p1_want(i) / #i 3.14 /
          p2_want(i,j) / i1.j1 3.14, i1.j2 3.14 /
          p2_want_merge(i,j) / i1.j1 3.14, i1.j2 3.14, i1.j3 3.14 /;

Variable v0      / fx 3.14 /
         v1(i)   / #i.fx 3.14 /
         v2(i,j) / #i.#j.fx 3.14 /
         v0_want / l 3.14, m 0, lo 0, up 10, scale 1 /
         v1_want(i) / i1.(l 3.14, m 0, lo 0, up 10, scale 1),
                      i2.(l 3.14, m 0, lo 0, up 10, scale 1) /
         v2_want(i,j) / i1.j1.(l 3.14, m 0, lo 0, up 10, scale 1),
                        i1.j2.(l 3.14, m 0, lo 0, up 10, scale 1) /
         v2_want_merge(i,j) / i1.j1.(l 3.14, m 0, lo 0, up 10, scale 1),
                              i1.j2.(l 3.14, m 0, lo 0, up 10, scale 1),
                              i1.j3.(l 3.14, m 0, lo 0, up 10, scale 1) /;

equation e0      / fx 3.14 /
         e1(i)   / #i.fx 3.14 /
         e2(i,j) / #i.#j.fx 3.14 /
         e0_want / l 3.14, m 0, lo 0, up 10, scale 1 /
         e1_want(i) / i1.(l 3.14, m 0, lo 0, up 10, scale 1),
                      i2.(l 3.14, m 0, lo 0, up 10, scale 1) /
         e2_want(i,j) / i1.j1.(l 3.14, m 0, lo 0, up 10, scale 1),
                        i1.j2.(l 3.14, m 0, lo 0, up 10, scale 1) /
         e2_want_merge(i,j) / i1.j1.(l 3.14, m 0, lo 0, up 10, scale 1),
                              i1.j2.(l 3.14, m 0, lo 0, up 10, scale 1),
                              i1.j3.(l 3.14, m 0, lo 0, up 10, scale 1) /;

$call rm -f embpy01_log.txt
$onEchoV > checkData
$gdxOut have.gdx
$unLoad %1
$gdxOut
$call gdxdiff wanted.gdx have.gdx >> embpy01_log.txt
$if errorlevel 1 $abort %2
$offEcho

*** Test that an empty code block works

$onEmbeddedCode Python:
$offEmbeddedCode

*** Test that an code block with comments only works

$onEmbeddedCode Python:
#
$offEmbeddedCode


$onEmbeddedCode Python:
#
  #
$offEmbeddedCode

$onEmbeddedCode Python:
#

  #
$offEmbeddedCode

*** Test default list behavior ***

*Test 1.1: Test default list behavior of scalars
$onEmbeddedCode Python:
  p0 = list(gams.get('p0'))
  assert p0 == [3.14], \
  "Test 1.1 failed in Python"
$offEmbeddedCode

*Test 1.2: Test default list behavior of one dimensional parameters
$onEmbeddedCode Python:
  p1 = list(gams.get('p1'))
  assert p1 == [('i1', 3.14), ('i2', 3.14)], \
  "Test 1.2 failed in Python"
$offEmbeddedCode

*Test 1.3: Test default list behavior of multi dimensional parameters
$onEmbeddedCode Python:
  p2 = list(gams.get('p2'))
  assert p2 == [(('i1', 'j1'), 3.14), \
                (('i1', 'j2'), 3.14), \
                (('i1', 'j3'), 3.14), \
                (('i2', 'j1'), 3.14), \
                (('i2', 'j2'), 3.14), \
                (('i2', 'j3'), 3.14)], \
  "Test 1.3 failed in Python"
$offEmbeddedCode

*Test 1.4: Test default list behavior of one dimensional sets
$onEmbeddedCode Python:
  i = list(gams.get('i'))
  assert i == ['i1', 'i2'], \
  "Test 1.4 failed in Python"
$offEmbeddedCode

*Test 1.5: Test default list behavior of multi dimensional sets
$onEmbeddedCode Python:
  ij = list(gams.get('ij'))
  assert ij == [('i1', 'j1'), ('i1', 'j2'), \
                ('i1', 'j3'), ('i2', 'j1'), \
                ('i2', 'j2'), ('i2', 'j3')], \
  "Test 1.5 failed in Python"
$offEmbeddedCode

*Test 1.6: Test default list behavior of scalar variables
$onEmbeddedCode Python:
  v0 = list(gams.get('v0'))
  assert v0 == [(3.14, 0, 3.14, 3.14, 1)], \
  "Test 1.6 failed in Python"
$offEmbeddedCode

*Test 1.7: Test default list behavior of one dimensional variables
$onEmbeddedCode Python:
  v1 = list(gams.get('v1'))
  assert v1 == [('i1', (3.14, 0, 3.14, 3.14, 1)), \
                ('i2', (3.14, 0, 3.14, 3.14, 1))], \
  "Test 1.7 failed in Python"
$offEmbeddedCode

*Test 1.8: Test default list behavior of multi dimensional variables
$onEmbeddedCode Python:
  v2 = list(gams.get('v2'))
  assert v2 == [(('i1', 'j1'), (3.14, 0, 3.14, 3.14, 1)), \
                (('i1', 'j2'), (3.14, 0, 3.14, 3.14, 1)), \
                (('i1', 'j3'), (3.14, 0, 3.14, 3.14, 1)), \
                (('i2', 'j1'), (3.14, 0, 3.14, 3.14, 1)), \
                (('i2', 'j2'), (3.14, 0, 3.14, 3.14, 1)), \
                (('i2', 'j3'), (3.14, 0, 3.14, 3.14, 1))], \
  "Test 1.8 failed in Python"
$offEmbeddedCode

*Test 1.9: Test default list behavior of scalar equations
$onEmbeddedCode Python:
  e0 = list(gams.get('e0'))
  assert e0 == [(3.14, 0, 3.14, 3.14, 1)], \
  "Test 1.9 failed in Python"
$offEmbeddedCode

*Test 1.10: Test default list behavior of one dimensional equations
$onEmbeddedCode Python:
  e1 = list(gams.get('e1'))
  assert e1 == [('i1', (3.14, 0, 3.14, 3.14, 1)), \
                ('i2', (3.14, 0, 3.14, 3.14, 1))], \
  "Test 1.10 failed in Python"
$offEmbeddedCode

*Test 1.11: Test default list behavior of multi dimensional equations
$onEmbeddedCode Python:
  e2 = list(gams.get('e2'))
  assert e2 == [(('i1', 'j1'), (3.14, 0, 3.14, 3.14, 1)), \
                (('i1', 'j2'), (3.14, 0, 3.14, 3.14, 1)), \
                (('i1', 'j3'), (3.14, 0, 3.14, 3.14, 1)), \
                (('i2', 'j1'), (3.14, 0, 3.14, 3.14, 1)), \
                (('i2', 'j2'), (3.14, 0, 3.14, 3.14, 1)), \
                (('i2', 'j3'), (3.14, 0, 3.14, 3.14, 1)) ], \
  "Test 1.11 failed in Python"
$offEmbeddedCode

*** Test different aspects of list behavior ***

*Test 2.1: Test list behavior of explanatory text
$onEmbeddedCode Python:
  i = list(gams.get('i', valueFormat=ValueFormat.FLAT))
  assert i == [('i1', 'text for i1'), ('i2', 'text for i2')], \
  "Test 2.1a failed in Python"
$offEmbeddedCode

*Test 2.1: Test list behavior of explanatory text
$onEmbeddedCode Python:
  i = list(gams.get('i', valueFormat=ValueFormat.TUPLE))
  assert i == [('i1', ('text for i1',)), ('i2', ('text for i2',))], \
  "Test 2.1b failed in Python"
$offEmbeddedCode

*** Test default list behavior using label indexes instead of strings ***

*Test 3.1: Test list behavior of scalars using label indexes
$onEmbeddedCode Python:
  p0 = list(gams.get('p0', KeyType.INT))
  assert p0 == [3.14], \
  "Test 3.1 failed in Python"
$offEmbeddedCode

*Test 3.2: Test list behavior of one dimensional parameters using label indexes
$onEmbeddedCode Python:
  p1 = list(gams.get('p1', KeyType.INT))
  assert p1 == [(1, 3.14), (2, 3.14)], \
  "Test 3.2 failed in Python"
$offEmbeddedCode

*Test 3.3: Test list behavior of multi dimensional parameters using label indexes
$onEmbeddedCode Python:
  p2 = list(gams.get('p2', KeyType.INT))
  assert p2 == [((1, 3), 3.14), \
                ((1, 4), 3.14), \
                ((1, 5), 3.14), \
                ((2, 3), 3.14), \
                ((2, 4), 3.14), \
                ((2, 5), 3.14)], \
  "Test 3.3 failed in Python"
$offEmbeddedCode

*Test 3.4: Test list behavior of one dimensional sets using label indexes
$onEmbeddedCode Python:
  i = list(gams.get('i', KeyType.INT))
  assert i == [1, 2], \
  "Test 3.4 failed in Python"
$offEmbeddedCode

*Test 3.5: Test list behavior of multi dimensional sets using label indexes
$onEmbeddedCode Python:
  ij = list(gams.get('ij', KeyType.INT))
  assert ij == [(1, 3), (1, 4), \
                (1, 5), (2, 3), \
                (2, 4), (2, 5)], \
  "Test 3.5 failed in Python"
$offEmbeddedCode

*Test 3.6: Test list behavior of scalar variables using label indexes
$onEmbeddedCode Python:
  v0 = list(gams.get('v0', KeyType.INT))
  assert v0 == [(3.14, 0, 3.14, 3.14, 1)], \
  "Test 3.6 failed in Python"
$offEmbeddedCode

*Test 3.7: Test list behavior of one dimensional variables using label indexes
$onEmbeddedCode Python:
  v1 = list(gams.get('v1', KeyType.INT))
  assert v1 == [(1, (3.14, 0, 3.14, 3.14, 1)), \
                (2, (3.14, 0, 3.14, 3.14, 1))], \
  "Test 3.7 failed in Python"
$offEmbeddedCode

*Test 3.8: Test list behavior of multi dimensional variables using label indexes
$onEmbeddedCode Python:
  v2 = list(gams.get('v2', KeyType.INT))
  assert v2 == [((1, 3), (3.14, 0, 3.14, 3.14, 1)), \
                ((1, 4), (3.14, 0, 3.14, 3.14, 1)), \
                ((1, 5), (3.14, 0, 3.14, 3.14, 1)), \
                ((2, 3), (3.14, 0, 3.14, 3.14, 1)), \
                ((2, 4), (3.14, 0, 3.14, 3.14, 1)), \
                ((2, 5), (3.14, 0, 3.14, 3.14, 1))], \
  "Test 3.8 failed in Python"
$offEmbeddedCode

*Test 3.9: Test list behavior of scalar equations using label indexes
$onEmbeddedCode Python:
  e0 = list(gams.get('e0', KeyType.INT))
  assert e0 == [(3.14, 0, 3.14, 3.14, 1)], \
  "Test 3.9 failed in Python"
$offEmbeddedCode

*Test 3.10: Test list behavior of one dimensional equations using label indexes
$onEmbeddedCode Python:
  e1 = list(gams.get('e1', KeyType.INT))
  assert e1 == [(1, (3.14, 0, 3.14, 3.14, 1)), \
                (2, (3.14, 0, 3.14, 3.14, 1))], \
  "Test 3.10 failed in Python"
$offEmbeddedCode

*Test 3.11: Test list behavior of multi dimensional equations using label indexes
$onEmbeddedCode Python:
  e2 = list(gams.get('e2', KeyType.INT))
  assert e2 == [((1, 3), (3.14, 0, 3.14, 3.14, 1)), \
                ((1, 4), (3.14, 0, 3.14, 3.14, 1)), \
                ((1, 5), (3.14, 0, 3.14, 3.14, 1)), \
                ((2, 3), (3.14, 0, 3.14, 3.14, 1)), \
                ((2, 4), (3.14, 0, 3.14, 3.14, 1)), \
                ((2, 5), (3.14, 0, 3.14, 3.14, 1)) ], \
  "Test 3.11 failed in Python"
$offEmbeddedCode


*** Test writing symbol data using labels ***
$onMultiR
*Test 4.1: Test writing of scalars
$gdxOut wanted.gdx
$unLoad p0_want=p0
$gdxOut

$clear p0
$onEmbeddedCode Python:
  data = [3.14]
  gams.set('p0', data)
$offEmbeddedCode p0
$batInclude checkData p0 'Test 4.1a failed in GAMS'

$clear p0
$onEmbeddedCode Python:
  data = [(3.14,)]
  gams.set('p0', data)
$offEmbeddedCode p0
$batInclude checkData p0 'Test 4.1b failed in GAMS'

*Test 4.2: Test writing of one dimensional parameters
$gdxOut wanted.gdx
$unLoad p1_want=p1
$gdxOut

$clear p1
$onEmbeddedCode Python:
  data = [("i1", 3.14), ("i2", 3.14)]
  gams.set('p1', data)
$offEmbeddedCode p1
$batInclude checkData p1 'Test 4.2a failed in GAMS'

$clear p1
$onEmbeddedCode Python:
  data = [(("i1",), 3.14), (("i2",), 3.14)]
  gams.set('p1', data)
$offEmbeddedCode p1
$batInclude checkData p1 'Test 4.2b failed in GAMS'

$clear p1
$onEmbeddedCode Python:
  data = [("i1", (3.14,)), ("i2", (3.14,))]
  gams.set('p1', data)
$offEmbeddedCode p1
$batInclude checkData p1 'Test 4.2c failed in GAMS'

$clear p1
$onEmbeddedCode Python:
  data = [(("i1",), (3.14,)), (("i2",), (3.14,))]
  gams.set('p1', data)
$offEmbeddedCode p1
$batInclude checkData p1 'Test 4.2d failed in GAMS'

*Test 4.3: Test writing of multi dimensional parameters
$gdxOut wanted.gdx
$unLoad p2_want=p2
$gdxOut

$clear p2
$onEmbeddedCode Python:
  data = [('i1', 'j1', 3.14), ('i1', 'j2', 3.14)]
  gams.set('p2', data)
$offEmbeddedCode p2
$batInclude checkData p2 'Test 4.3a failed in GAMS'

$clear p2
$onEmbeddedCode Python:
  data = [(('i1', 'j1'), 3.14), (('i1', 'j2'), 3.14)]
  gams.set('p2', data)
$offEmbeddedCode p2
$batInclude checkData p2 'Test 4.3b failed in GAMS'

$clear p2
$onEmbeddedCode Python:
  data = [('i1', 'j1', (3.14,)), ('i1', 'j2', (3.14,))]
  gams.set('p2', data)
$offEmbeddedCode p2
$batInclude checkData p2 'Test 4.3c failed in GAMS'

$clear p2
$onEmbeddedCode Python:
  data = [(('i1', 'j1'), (3.14,)), (('i1', 'j2'), (3.14,))]
  gams.set('p2', data)
$offEmbeddedCode p2
$batInclude checkData p2 'Test 4.3d failed in GAMS'

*Test 4.4: Test writing of one dimensional sets
$gdxOut wanted.gdx
$unLoad i_want=iout
$gdxOut

$clear iout
$onEmbeddedCode Python:
  data = ['i1', 'i2']
  gams.set('iout', data)
$offEmbeddedCode iout
$batInclude checkData iout 'Test 4.4a failed in GAMS'

$clear iout
$onEmbeddedCode Python:
  data = [('i1',), ('i2',)]
  gams.set('iout', data)
$offEmbeddedCode iout
$batInclude checkData iout 'Test 4.4b failed in GAMS'

$gdxOut wanted.gdx
$unLoad i_want_ex=iout
$gdxOut

$clear iout
$onEmbeddedCode Python:
  data = [('i1', "text 1"), ('i2', "text 2")]
  gams.set('iout', data)
$offEmbeddedCode iout
$batInclude checkData iout 'Test 4.4c failed in GAMS'

$clear iout
$onEmbeddedCode Python:
  data = [(('i1',), "text 1"), (('i2',), "text 2")]
  gams.set('iout', data)
$offEmbeddedCode iout
$batInclude checkData iout 'Test 4.4d failed in GAMS'

$clear iout
$onEmbeddedCode Python:
  data = [('i1', ("text 1",)), ('i2', ("text 2",))]
  gams.set('iout', data)
$offEmbeddedCode iout
$batInclude checkData iout 'Test 4.4e failed in GAMS'

$clear iout
$onEmbeddedCode Python:
  data = [(('i1',), ("text 1",)), (('i2',), ("text 2",))]
  gams.set('iout', data)
$offEmbeddedCode iout
$batInclude checkData iout 'Test 4.4f failed in GAMS'

*Test 4.5: Test writing of multi dimensional sets
$gdxOut wanted.gdx
$unLoad ij_want=ij
$gdxOut

$clear ij
$onEmbeddedCode Python:
  data = [('i1', 'j1'), ('i1', 'j2')]
  gams.set('ij', data)
$offEmbeddedCode ij
$batInclude checkData ij 'Test 4.5a failed in GAMS'

$clear ij
$onEmbeddedCode Python:
  data = [(('i1', 'j1'),), (('i1', 'j2'),)]
  gams.set('ij', data)
$offEmbeddedCode ij
$batInclude checkData ij 'Test 4.5b failed in GAMS'

$gdxOut wanted.gdx
$unLoad ij_want_ex=ij
$gdxOut

$clear ij
$onEmbeddedCode Python:
  data = [('i1', 'j1', 'text 1'), ('i1', 'j2', 'text 2')]
  gams.set('ij', data)
$offEmbeddedCode ij
$batInclude checkData ij 'Test 4.5c failed in GAMS'

$clear ij
$onEmbeddedCode Python:
  data = [(('i1', 'j1'), 'text 1'), (('i1', 'j2'), 'text 2')]
  gams.set('ij', data)
$offEmbeddedCode ij
$batInclude checkData ij 'Test 4.5d failed in GAMS'

$clear ij
$onEmbeddedCode Python:
  data = [('i1', 'j1', ('text 1',)), ('i1', 'j2', ('text 2',))]
  gams.set('ij', data)
$offEmbeddedCode ij
$batInclude checkData ij 'Test 4.5e failed in GAMS'

$clear ij
$onEmbeddedCode Python:
  data = [(('i1', 'j1'), ('text 1',)), (('i1', 'j2'), ('text 2',))]
  gams.set('ij', data)
$offEmbeddedCode ij
$batInclude checkData ij 'Test 4.5f failed in GAMS'

*Test 4.6: Test writing of scalar variable
$gdxOut wanted.gdx
$unLoad v0_want=v0
$gdxOut

$clear v0
$onEmbeddedCode Python:
  data = [(3.14, 0, 0, 10, 1)]
  gams.set('v0', data)
$offEmbeddedCode v0
$batInclude checkData v0 'Test 4.6a failed in GAMS'

*Test 4.7: Test writing of one dimensonal variables
$gdxOut wanted.gdx
$unLoad v1_want=v1
$gdxOut

$clear v1
$onEmbeddedCode Python:
  data = [("i1", 3.14, 0, 0, 10, 1), ("i2", 3.14, 0, 0, 10, 1)]
  gams.set('v1', data)
$offEmbeddedCode v1
$batInclude checkData v1 'Test 4.7a failed in GAMS'

$clear v1
$onEmbeddedCode Python:
  data = [(("i1",), 3.14, 0, 0, 10, 1), (("i2",), 3.14, 0, 0, 10, 1)]
  gams.set('v1', data)
$offEmbeddedCode v1
$batInclude checkData v1 'Test 4.7b failed in GAMS'

$clear v1
$onEmbeddedCode Python:
  data = [("i1", (3.14, 0, 0, 10, 1)), ("i2", (3.14, 0, 0, 10, 1))]
  gams.set('v1', data)
$offEmbeddedCode v1
$batInclude checkData v1 'Test 4.7c failed in GAMS'

$clear v1
$onEmbeddedCode Python:
  data = [(("i1",), (3.14, 0, 0, 10, 1)), (("i2",), (3.14, 0, 0, 10, 1))]
  gams.set('v1', data)
$offEmbeddedCode v1
$batInclude checkData v1 'Test 4.7d failed in GAMS'

*Test 4.8: Test writing of multi dimensonal variables
$gdxOut wanted.gdx
$unLoad v2_want=v2
$gdxOut

$clear v2
$onEmbeddedCode Python:
  data = [("i1", "j1", 3.14, 0, 0, 10, 1), ("i1", "j2", 3.14, 0, 0, 10, 1)]
  gams.set('v2', data)
$offEmbeddedCode v2
$batInclude checkData v2 'Test 4.8a failed in GAMS'

$clear v2
$onEmbeddedCode Python:
  data = [(("i1", "j1"), 3.14, 0, 0, 10, 1), (("i1", "j2"), 3.14, 0, 0, 10, 1)]
  gams.set('v2', data)
$offEmbeddedCode v2
$batInclude checkData v2 'Test 4.8b failed in GAMS'

$clear v2
$onEmbeddedCode Python:
  data = [("i1", "j1", (3.14, 0, 0, 10, 1)), ("i1", "j2", (3.14, 0, 0, 10, 1))]
  gams.set('v2', data)
$offEmbeddedCode v2
$batInclude checkData v2 'Test 4.8c failed in GAMS'

$clear v2
$onEmbeddedCode Python:
  data = [(("i1", "j1"), (3.14, 0, 0, 10, 1)), (("i1", "j2"), (3.14, 0, 0, 10, 1))]
  gams.set('v2', data)
$offEmbeddedCode v2
$batInclude checkData v2 'Test 4.8d failed in GAMS'

*Test 4.9: Test writing of scalar equations
$gdxOut wanted.gdx
$unLoad e0_want=e0
$gdxOut

$clear e0
$onEmbeddedCode Python:
  data = [(3.14, 0, 0, 10, 1)]
  gams.set('e0', data)
$offEmbeddedCode e0
$batInclude checkData e0 'Test 4.9 failed in GAMS'


*Test 4.10: Test writing of one dimensonal equations
$gdxOut wanted.gdx
$unLoad e1_want=e1
$gdxOut

$clear e1
$onEmbeddedCode Python:
  data = [("i1", 3.14, 0, 0, 10, 1), ("i2", 3.14, 0, 0, 10, 1)]
  gams.set('e1', data)
$offEmbeddedCode e1
$batInclude checkData e1 'Test 4.10a failed in GAMS'

$clear e1
$onEmbeddedCode Python:
  data = [(("i1",), 3.14, 0, 0, 10, 1), (("i2",), 3.14, 0, 0, 10, 1)]
  gams.set('e1', data)
$offEmbeddedCode e1
$batInclude checkData e1 'Test 4.10b failed in GAMS'

$clear e1
$onEmbeddedCode Python:
  data = [("i1", (3.14, 0, 0, 10, 1)), ("i2", (3.14, 0, 0, 10, 1))]
  gams.set('e1', data)
$offEmbeddedCode e1
$batInclude checkData e1 'Test 4.10c failed in GAMS'

$clear e1
$onEmbeddedCode Python:
  data = [(("i1",), (3.14, 0, 0, 10, 1)), (("i2",), (3.14, 0, 0, 10, 1))]
  gams.set('e1', data)
$offEmbeddedCode e1
$batInclude checkData e1 'Test 4.10d failed in GAMS'

*Test 4.11: Test writing of multi dimensonal equations
$gdxOut wanted.gdx
$unLoad e2_want=e2
$gdxOut

$clear e2
$onEmbeddedCode Python:
  data = [("i1", "j1", 3.14, 0, 0, 10, 1), ("i1", "j2", 3.14, 0, 0, 10, 1)]
  gams.set('e2', data)
$offEmbeddedCode e2
$batInclude checkData e2 'Test 4.11a failed in GAMS'

$clear e2
$onEmbeddedCode Python:
  data = [(("i1", "j1"), 3.14, 0, 0, 10, 1), (("i1", "j2"), 3.14, 0, 0, 10, 1)]
  gams.set('e2', data)
$offEmbeddedCode e2
$batInclude checkData e2 'Test 4.11b failed in GAMS'

$clear e2
$onEmbeddedCode Python:
  data = [("i1", "j1", (3.14, 0, 0, 10, 1)), ("i1", "j2", (3.14, 0, 0, 10, 1))]
  gams.set('e2', data)
$offEmbeddedCode e2
$batInclude checkData e2 'Test 4.11c failed in GAMS'

$clear e2
$onEmbeddedCode Python:
  data = [(("i1", "j1"), (3.14, 0, 0, 10, 1)), (("i1", "j2"), (3.14, 0, 0, 10, 1))]
  gams.set('e2', data)
$offEmbeddedCode e2
$batInclude checkData e2 'Test 4.11d failed in GAMS'


*** Test writing symbol data using label indexes instead of strings ***

*Test 5.1: Test writing of scalars using label indexes
$gdxOut wanted.gdx
$unLoad p0_want=p0
$gdxOut

$clear p0
$onEmbeddedCode Python:
  data = [3.14]
  gams.set('p0', data)
$offEmbeddedCode p0
$batInclude checkData p0 'Test 5.1a failed in GAMS'

$clear p0
$onEmbeddedCode Python:
  data = [(3.14,)]
  gams.set('p0', data)
$offEmbeddedCode p0
$batInclude checkData p0 'Test 5.1b failed in GAMS'

*Test 5.2: Test writing of one dimensional parameters using label indexes
$gdxOut wanted.gdx
$unLoad p1_want=p1
$gdxOut

$clear p1
$onEmbeddedCode Python:
  data = [(1, 3.14), (2, 3.14)]
  gams.set('p1', data)
$offEmbeddedCode p1
$batInclude checkData p1 'Test 5.2a failed in GAMS'

$clear p1
$onEmbeddedCode Python:
  data = [((1,), 3.14), ((2,), 3.14)]
  gams.set('p1', data)
$offEmbeddedCode p1
$batInclude checkData p1 'Test 5.2b failed in GAMS'

$clear p1
$onEmbeddedCode Python:
  data = [(1, (3.14,)), (2, (3.14,))]
  gams.set('p1', data)
$offEmbeddedCode p1
$batInclude checkData p1 'Test 5.2c failed in GAMS'

$clear p1
$onEmbeddedCode Python:
  data = [((1,), (3.14,)), ((2,), (3.14,))]
  gams.set('p1', data)
$offEmbeddedCode p1
$batInclude checkData p1 'Test 5.2d failed in GAMS'

*Test 5.3: Test writing of multi dimensional parameters using label indexes
$gdxOut wanted.gdx
$unLoad p2_want=p2
$gdxOut

$clear p2
$onEmbeddedCode Python:
  data = [(1, 3, 3.14), (1, 4, 3.14)]
  gams.set('p2', data)
$offEmbeddedCode p2
$batInclude checkData p2 'Test 5.3a failed in GAMS'

$clear p2
$onEmbeddedCode Python:
  data = [((1, 3), 3.14), ((1, 4), 3.14)]
  gams.set('p2', data)
$offEmbeddedCode p2
$batInclude checkData p2 'Test 5.3b failed in GAMS'

$clear p2
$onEmbeddedCode Python:
  data = [(1, 3, (3.14,)), (1, 4, (3.14,))]
  gams.set('p2', data)
$offEmbeddedCode p2
$batInclude checkData p2 'Test 5.3c failed in GAMS'

$clear p2
$onEmbeddedCode Python:
  data = [((1, 3), (3.14,)), ((1, 4), (3.14,))]
  gams.set('p2', data)
$offEmbeddedCode p2
$batInclude checkData p2 'Test 5.3d failed in GAMS'



*Test 5.4: Test writing of one dimensional sets using label indexes
$gdxOut wanted.gdx
$unLoad i_want=iout
$gdxOut

$clear iout
$onEmbeddedCode Python:
  data = [1, 2]
  gams.set('iout', data)
$offEmbeddedCode iout
$batInclude checkData iout 'Test 5.4a failed in GAMS'

$clear iout
$onEmbeddedCode Python:
  data = [(1,), (2,)]
  gams.set('iout', data)
$offEmbeddedCode iout
$batInclude checkData iout 'Test 5.4b failed in GAMS'

$gdxOut wanted.gdx
$unLoad i_want_ex=iout
$gdxOut

$clear iout
$onEmbeddedCode Python:
  data = [(1, "text 1"), (2, "text 2")]
  gams.set('iout', data)
$offEmbeddedCode iout
$batInclude checkData iout 'Test 5.4c failed in GAMS'

$clear iout
$onEmbeddedCode Python:
  data = [((1,), "text 1"), ((2,), "text 2")]
  gams.set('iout', data)
$offEmbeddedCode iout
$batInclude checkData iout 'Test 5.4d failed in GAMS'

$clear iout
$onEmbeddedCode Python:
  data = [(1, ("text 1",)), (2, ("text 2",))]
  gams.set('iout', data)
$offEmbeddedCode iout
$batInclude checkData iout 'Test 5.4e failed in GAMS'

$clear iout
$onEmbeddedCode Python:
  data = [((1,), ("text 1",)), ((2,), ("text 2",))]
  gams.set('iout', data)
$offEmbeddedCode iout
$batInclude checkData iout 'Test 5.4f failed in GAMS'

*Test 5.5: Test writing of multi dimensional sets using label indexes
$gdxOut wanted.gdx
$unLoad ij_want=ij
$gdxOut

$clear ij
$onEmbeddedCode Python:
  data = [(1, 3), (1, 4)]
  gams.set('ij', data)
$offEmbeddedCode ij
$batInclude checkData ij 'Test 5.5a failed in GAMS'

$clear ij
$onEmbeddedCode Python:
  data = [((1, 3),), ((1, 4),)]
  gams.set('ij', data)
$offEmbeddedCode ij
$batInclude checkData ij 'Test 5.5b failed in GAMS'

$gdxOut wanted.gdx
$unLoad ij_want_ex=ij
$gdxOut

$clear ij
$onEmbeddedCode Python:
  data = [(1, 3, 'text 1'), (1, 4, 'text 2')]
  gams.set('ij', data)
$offEmbeddedCode ij
$batInclude checkData ij 'Test 5.5c failed in GAMS'

$clear ij
$onEmbeddedCode Python:
  data = [((1, 3), 'text 1'), ((1, 4), 'text 2')]
  gams.set('ij', data)
$offEmbeddedCode ij
$batInclude checkData ij 'Test 5.5d failed in GAMS'

$clear ij
$onEmbeddedCode Python:
  data = [(1, 3, ('text 1',)), (1, 4, ('text 2',))]
  gams.set('ij', data)
$offEmbeddedCode ij
$batInclude checkData ij 'Test 5.5e failed in GAMS'

$clear ij
$onEmbeddedCode Python:
  data = [((1, 3), ('text 1',)), ((1, 4), ('text 2',))]
  gams.set('ij', data)
$offEmbeddedCode ij
$batInclude checkData ij 'Test 5.5f failed in GAMS'

*Test 5.6: Test writing of scalar variable using label indexes
$gdxOut wanted.gdx
$unLoad v0_want=v0
$gdxOut

$clear v0
$onEmbeddedCode Python:
  data = [(3.14, 0, 0, 10, 1)]
  gams.set('v0', data)
$offEmbeddedCode v0
$batInclude checkData v0 'Test 5.6a failed in GAMS'

*Test 5.7: Test writing of one dimensonal variables using label indexes
$gdxOut wanted.gdx
$unLoad v1_want=v1
$gdxOut

$onEmbeddedCode Python:
  data = [(1, 3.14, 0, 0, 10, 1), (2, 3.14, 0, 0, 10, 1)]
  gams.set('v1', data)
$offEmbeddedCode v1
$batInclude checkData v1 'Test 5.7a failed in GAMS'

$clear v1
$onEmbeddedCode Python:
  data = [((1,), 3.14, 0, 0, 10, 1), ((2,), 3.14, 0, 0, 10, 1)]
  gams.set('v1', data)
$offEmbeddedCode v1
$batInclude checkData v1 'Test 5.7b failed in GAMS'

$clear v1
$onEmbeddedCode Python:
  data = [(1, (3.14, 0, 0, 10, 1)), (2, (3.14, 0, 0, 10, 1))]
  gams.set('v1', data)
$offEmbeddedCode v1
$batInclude checkData v1 'Test 5.7c failed in GAMS'

$clear v1
$onEmbeddedCode Python:
  data = [((1,), (3.14, 0, 0, 10, 1)), ((2,), (3.14, 0, 0, 10, 1))]
  gams.set('v1', data)
$offEmbeddedCode v1
$batInclude checkData v1 'Test 5.7d failed in GAMS'

*Test 5.8: Test writing of multi dimensonal variables using label indexes
$gdxOut wanted.gdx
$unLoad v2_want=v2
$gdxOut

$clear v2
$onEmbeddedCode Python:
  data = [(1, 3, 3.14, 0, 0, 10, 1), (1, 4, 3.14, 0, 0, 10, 1)]
  gams.set('v2', data)
$offEmbeddedCode v2
$batInclude checkData v2 'Test 5.8a failed in GAMS'

$clear v2
$onEmbeddedCode Python:
  data = [((1, 3), 3.14, 0, 0, 10, 1), ((1, 4), 3.14, 0, 0, 10, 1)]
  gams.set('v2', data)
$offEmbeddedCode v2
$batInclude checkData v2 'Test 5.8b failed in GAMS'

$clear v2
$onEmbeddedCode Python:
  data = [(1, 3, (3.14, 0, 0, 10, 1)), (1, 4, (3.14, 0, 0, 10, 1))]
  gams.set('v2', data)
$offEmbeddedCode v2
$batInclude checkData v2 'Test 5.8c failed in GAMS'

$clear v2
$onEmbeddedCode Python:
  data = [((1, 3), (3.14, 0, 0, 10, 1)), ((1, 4), (3.14, 0, 0, 10, 1))]
  gams.set('v2', data)
$offEmbeddedCode v2
$batInclude checkData v2 'Test 5.8d failed in GAMS'

*Test 5.9: Test writing of scalar equations using label indexes
$gdxOut wanted.gdx
$unLoad e0_want=e0
$gdxOut

$clear e0
$onEmbeddedCode Python:
  data = [(3.14, 0, 0, 10, 1)]
  gams.set('e0', data)
$offEmbeddedCode e0
$batInclude checkData e0 'Test 5.9 failed in GAMS'

*Test 5.10: Test writing of one dimensonal equations using label indexes
$gdxOut wanted.gdx
$unLoad e1_want=e1
$gdxOut

$clear e1
$onEmbeddedCode Python:
  data = [(1, 3.14, 0, 0, 10, 1), (2, 3.14, 0, 0, 10, 1)]
  gams.set('e1', data)
$offEmbeddedCode e1
$batInclude checkData e1 'Test 5.10a failed in GAMS'

$clear e1
$onEmbeddedCode Python:
  data = [((1,), 3.14, 0, 0, 10, 1), ((2,), 3.14, 0, 0, 10, 1)]
  gams.set('e1', data)
$offEmbeddedCode e1
$batInclude checkData e1 'Test 5.10b failed in GAMS'

$clear e1
$onEmbeddedCode Python:
  data = [(1, (3.14, 0, 0, 10, 1)), (2, (3.14, 0, 0, 10, 1))]
  gams.set('e1', data)
$offEmbeddedCode e1
$batInclude checkData e1 'Test 5.10c failed in GAMS'

$clear e1
$onEmbeddedCode Python:
  data = [((1,), (3.14, 0, 0, 10, 1)), ((2,), (3.14, 0, 0, 10, 1))]
  gams.set('e1', data)
$offEmbeddedCode e1
$batInclude checkData e1 'Test 5.10d failed in GAMS'

*Test 5.11: Test writing of multi dimensonal equations using label indexes
$gdxOut wanted.gdx
$unLoad e2_want=e2
$gdxOut

$clear e2
$onEmbeddedCode Python:
  data = [(1, 3, 3.14, 0, 0, 10, 1), (1, 4, 3.14, 0, 0, 10, 1)]
  gams.set('e2', data)
$offEmbeddedCode e2
$batInclude checkData e2 'Test 5.11a failed in GAMS'

$clear e2
$onEmbeddedCode Python:
  data = [((1, 3), 3.14, 0, 0, 10, 1), ((1, 4), 3.14, 0, 0, 10, 1)]
  gams.set('e2', data)
$offEmbeddedCode e2
$batInclude checkData e2 'Test 5.11b failed in GAMS'

$clear e2
$onEmbeddedCode Python:
  data = [(1, 3, (3.14, 0, 0, 10, 1)), (1, 4, (3.14, 0, 0, 10, 1))]
  gams.set('e2', data)
$offEmbeddedCode e2
$batInclude checkData e2 'Test 5.11c failed in GAMS'

$clear e2
$onEmbeddedCode Python:
  data = [((1, 3), (3.14, 0, 0, 10, 1)), ((1, 4), (3.14, 0, 0, 10, 1))]
  gams.set('e2', data)
$offEmbeddedCode e2
$batInclude checkData e2 'Test 5.11d failed in GAMS'


*** Test adding/retrieving labels/label indexes

*Test 6.1: Test retrieving label indexes from label
$onEmbeddedCode Python:
  label = gams.getUel(1)
  assert label == 'i1', \
  "Test 6.1 failed in Python"
  label = gams.getUel(3)
  assert label == 'j1', \
  "Test 6.1 failed in Python"
$offEmbeddedCode

*Test 6.2: Test merging/adding of labels
$onEmbeddedCode Python:
  nrUels = gams.getUelCount()
  assert nrUels == 5, \
  "Test 6.2 failed in Python"

  labelIndex = gams.mergeUel("i2")
  assert labelIndex == 2, \
  "Test 6.2 failed in Python"
  nrUels = gams.getUelCount()
  assert nrUels == 5, \
  "Test 6.2 failed in Python"

  labelIndex = gams.mergeUel("new1")
  assert labelIndex == 6, \
  "Test 6.2 failed in Python"
  nrUels = gams.getUelCount()
  assert nrUels == 6, \
  "Test 6.2 failed in Python"

  label = gams.getUel(6)
  assert label == 'new1', \
  "Test 6.2 failed in Python"
$offEmbeddedCode

*** Test set() method using mergeType=MergeType.MERGE using labels

*Test 7.1: Test mergeType=MergeType.MERGE with parameters
$gdxOut wanted.gdx
$unLoad p2_want_merge=p2
$gdxOut
$clear p2
$onEmbeddedCode Python:
  data = [('i1', 'j1', 3.14), ('i1', 'j2', 3.14)]
  gams.set('p2', data)
$offEmbeddedCode p2
$onEmbeddedCode Python:
  data = [('i1', 'j2', 3.14), ('i1', 'j3', 3.14)]
  gams.set('p2', data, mergeType=MergeType.MERGE)
$offEmbeddedCode p2
$batInclude checkData p2 'Test 7.1 failed in GAMS'

*Test 7.2: Test mergeType=MergeType.MERGE with sets
$gdxOut wanted.gdx
$unLoad ij_want_merge=ij
$gdxOut
$clear ij
$onEmbeddedCode Python:
  data = [('i1', 'j1'), ('i1', 'j2')]
  gams.set('ij', data)
$offEmbeddedCode ij
$onEmbeddedCode Python:
  data = [('i1', 'j2'), ('i1', 'j3')]
  gams.set('ij', data, mergeType=MergeType.MERGE)
$offEmbeddedCode ij
$batInclude checkData ij 'Test 7.2 failed in GAMS'

*Test 7.3: Test mergeType=MergeType.MERGE with variables
$gdxOut wanted.gdx
$unLoad v2_want_merge=v2
$gdxOut
$clear v2
$onEmbeddedCode Python:
  data = [("i1", "j1", 3.14, 0, 0, 10, 1), ("i1", "j2", 3.14, 0, 0, 10, 1)]
  gams.set('v2', data)
$offEmbeddedCode v2
$onEmbeddedCode Python:
  data = [("i1", "j2", 3.14, 0, 0, 10, 1), ("i1", "j3", 3.14, 0, 0, 10, 1)]
  gams.set('v2', data, mergeType=MergeType.MERGE)
$offEmbeddedCode v2
$batInclude checkData v2 'Test 7.3 failed in GAMS'

*Test 7.4: Test mergeType=MergeType.MERGE with equations
$gdxOut wanted.gdx
$unLoad e2_want_merge=e2
$gdxOut
$clear e2
$onEmbeddedCode Python:
  data = [("i1", "j1", 3.14, 0, 0, 10, 1), ("i1", "j2", 3.14, 0, 0, 10, 1)]
  gams.set('e2', data)
$offEmbeddedCode e2
$onEmbeddedCode Python:
  data = [("i1", "j2", 3.14, 0, 0, 10, 1), ("i1", "j3", 3.14, 0, 0, 10, 1)]
  gams.set('e2', data, mergeType=MergeType.MERGE)
$offEmbeddedCode e2
$batInclude checkData e2 'Test 7.4 failed in GAMS'


*** Test set() method using mergeType=MergeType.MERGE using label indexes instead of strings

*Test 8.1: Test mergeType=MergeType.MERGE with parameters using label indexes
$gdxOut wanted.gdx
$unLoad p2_want_merge=p2
$gdxOut
$clear p2
$onEmbeddedCode Python:
  data = [(1, 3, 3.14), (1, 4, 3.14)]
  gams.set('p2', data)
$offEmbeddedCode p2
$onEmbeddedCode Python:
  data = [(1, 4, 3.14), (1, 5, 3.14)]
  gams.set('p2', data, mergeType=MergeType.MERGE)
$offEmbeddedCode p2
$batInclude checkData p2 'Test 8.1 failed in GAMS'

*Test 8.2: Test mergeType=MergeType.MERGE with sets using label indexes
$gdxOut wanted.gdx
$unLoad ij_want_merge=ij
$gdxOut
$clear ij
$onEmbeddedCode Python:
  data = [(1, 3), (1, 4)]
  gams.set('ij', data)
$offEmbeddedCode ij
$onEmbeddedCode Python:
  data = [(1, 4), (1, 5)]
  gams.set('ij', data, mergeType=MergeType.MERGE)
$offEmbeddedCode ij
$batInclude checkData ij 'Test 8.2 failed in GAMS'

*Test 8.3: Test mergeType=MergeType.MERGE with variables using label indexes
$gdxOut wanted.gdx
$unLoad v2_want_merge=v2
$gdxOut
$clear v2
$onEmbeddedCode Python:
  data = [(1, 3, 3.14, 0, 0, 10, 1), (1, 4, 3.14, 0, 0, 10, 1)]
  gams.set('v2', data)
$offEmbeddedCode v2
$onEmbeddedCode Python:
  data = [(1, 4, 3.14, 0, 0, 10, 1), (1, 5, 3.14, 0, 0, 10, 1)]
  gams.set('v2', data, mergeType=MergeType.MERGE)
$offEmbeddedCode v2
$batInclude checkData v2 'Test 8.3 failed in GAMS'

*Test 8.4: Test mergeType=MergeType.MERGE with equations using label indexes
$gdxOut wanted.gdx
$unLoad e2_want_merge=e2
$gdxOut
$clear e2
$onEmbeddedCode Python:
  data = [(1, 3, 3.14, 0, 0, 10, 1), (1, 4, 3.14, 0, 0, 10, 1)]
  gams.set('e2', data)
$offEmbeddedCode e2
$onEmbeddedCode Python:
  data = [(1, 4, 3.14, 0, 0, 10, 1), (1, 5, 3.14, 0, 0, 10, 1)]
  gams.set('e2', data, mergeType=MergeType.MERGE)
$offEmbeddedCode e2
$batInclude checkData e2 'Test 8.4 failed in GAMS'


*** Miscellaneous tests

*Test 9.1: Test reading of empty symbols
$clear p2
$onEmbeddedCode Python:
  p2 = list(gams.get('p2'))
  assert p2 == [], \
  "Test 9.1 failed in Python"
$offEmbeddedCode

*Test 9.2: Test writing of empty symbols
$clear p2_want
$gdxOut wanted.gdx
$unLoad p2_want=p2
$gdxOut
$onEmbeddedCode Python:
  data = []
  gams.set('p2', data)
$offEmbeddedCode p2
$batInclude checkData p2 'Test 9.2 failed in GAMS'

*Test 9.3: Test writing of empty scalars
$clear p0_want
$gdxOut wanted.gdx
$unLoad p0_want=p0
$gdxOut
$onEmbeddedCode Python:
  data = []
  gams.set('p0', data)
$offEmbeddedCode p0
$batInclude checkData p0 'Test 9.3 failed in GAMS'


*Test 9.4: Test clear behavior of scalar symbols and their default values
$clear p0_want
$clear e0_want
positive variable xpos / l 0 /;
negative variable xneg / l 0 /;

$gdxOut wanted.gdx
$unLoad p0_want=p0 e0_want=e0 xpos xneg
$gdxOut

$onEmbeddedCode Python:
  gams.db['p0'].clear()
  gams.db['e0'].clear()
  gams.db['xpos'].clear()
  gams.db['xneg'].clear()
$offEmbeddedCode p0, e0, xpos, xneg
$batInclude checkData 'p0 e0 xpos xneg' 'Test 9.4 failed in GAMS'


*Test 9.5: Test special values
Set svIdx / myUNDF, myNA, myPINF, myMINF, myEPS /;
$onUNDF 
Parameter sv_want(svIdx) / myUNDF UNDF, myNA NA, myPINF INF, myMINF -INF, myEPS EPS /
          svX(svIdx), svY(svIdx) 

$onEmbeddedCode Python:
gams.epsAsZero = False
sv = dict(gams.get('sv_want'))
if sv['myUNDF']!=GMS_SV_UNDEF:
  raise ValueError('Test 9.5 failed in GAMS: Expect UNDEF')
if sv['myNA']==sv['myNA']:
  raise ValueError('Test 9.5 failed in GAMS: Expect NAN')
if sv['myPINF']!=float('inf'):
  raise ValueError('Test 9.5 failed in GAMS: Expect INF')
if sv['myMINF']!=float('-inf'):
  raise ValueError('Test 9.5 failed in GAMS: Expect -INF')
if sv['myEPS']!= gams._eps:
  raise ValueError('Test 9.5 failed in GAMS: Expect EPS')
gams.db['sv_want'].copy_symbol(gams.db['svX'])
gams.set('svY',[('myUNDF',GMS_SV_UNDEF),('myNA',float('nan')),('myPINF',float('inf')),('myMINF',float('-inf')),('myEPS',gams._eps)])
$offEmbeddedCode svX svY

$offUNDF

$gdxOut wanted.gdx
$unLoad sv_want=svX
$gdxOut
$batInclude checkData 'svX' 'Test 9.5 failed in GAMS (svX)'

$gdxOut wanted.gdx
$unLoad sv_want=svY
$gdxOut
$batInclude checkData 'svY' 'Test 9.5 failed in GAMS (svY)'


*Test 9.6: Test special values (EPS)
Parameter svZeroToEps(svIdx), svZeroToEps_want(svIdx) / myEPS EPS/;
Parameter svZeroToZero(svIdx), svZeroToZero_want(svIdx);

$onEps
$onEmbeddedCode Python:
gams.db['svZeroToEPS'].add_record('myEPS').value = 0
$offEmbeddedCode svZeroToEPS
$gdxOut wanted.gdx
$unLoad svZeroToEPS_want=svZeroToEPS
$gdxOut
$batInclude checkData 'svZeroToEPS' 'Test 9.6 failed in GAMS (svZeroToEPS)'

$offEps
$onEmbeddedCode Python:
gams.db['svZeroToZero'].add_record('myEPS').value = 0
$offEmbeddedCode svZeroToZero
$gdxOut wanted.gdx
$unLoad svZeroToZero_want=svZeroToZero
$gdxOut
$batInclude checkData 'svZeroToZero' 'Test 9.6 failed in GAMS (svZeroToZero)'

parameter witheps(*) / i1 1, i2 EPS /;
$onEmbeddedCode Python:
gams.ws.my_eps = 0
p = [ r.value for r in gams.db['witheps'] ]
assert p == [1.0, 0], \
  "Test 9.7 failed in Python"
$offEmbeddedCode

$onEmbeddedCode Python:
p = [ r.value for r in gams.db['witheps'] ]
assert p == [1.0, 4.94066E-324], \
  "Test 9.8 failed in Python"
$offEmbeddedCode

* Tests for long strings and auto indentation behavior

*Test 10.1: Test long strings : Skip of auto-indentation
$onEmbeddedCode Python:
s = """some
long
  string
""" + '''and another''' + """ long
string
  to
be added
"""
assert s == "some\nlong\n  string\nand another long\nstring\n  to\nbe added\n", \
"Test 10.1 failed in Python"
$offEmbeddedCode


*Test 10.2: Test long strings : Skip of auto-indentation
$onEmbeddedCode Python:
s = '''some
long
  string
''' + """and another""" + ''' long
string
  to
be added
'''
assert s == "some\nlong\n  string\nand another long\nstring\n  to\nbe added\n", \
"Test 10.2 failed in Python"
$offEmbeddedCode


*Test 10.3: Test long strings : embed """ in '''
$onEmbeddedCode Python:
s = '''some
long
  string
""" + """and another""" + """ long
string
  to
be added
'''
assert s == '''some\nlong\n  string\n""" + """and another""" + """ long\nstring\n  to\nbe added\n''', \
"Test 10.3 failed in Python"
$offEmbeddedCode


*Test 10.4: Test long strings : embed ''' in """
$onEmbeddedCode Python:
s = """some
long
  string
''' + '''and another''' + ''' long
string
  to
be added
"""
assert s == """some\nlong\n  string\n''' + '''and another''' + ''' long\nstring\n  to\nbe added\n""", \
"Test 10.4 failed in Python"
$offEmbeddedCode


*Test 10.5: Test behavior of line continuation character
$onEmbeddedCode Python:
s = "some string\
with line continuation\
characters"
assert s == "some string  with line continuation  characters", \
"Test 10.5 failed in Python"
$offEmbeddedCode


*Test 10.6: Mix line continuation character with long strings
$onEmbeddedCode Python:
s = """some string\
with line continuation
characters"""
assert s == "some stringwith line continuation\ncharacters", \
"Test 10.6 failed in Python"
$offEmbeddedCode

*Test 11: Run in a directory with long path name
$call mkdir 1234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890
$onEcho > 1234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890%system.dirsep%ec11.gms
$onEmbeddedCode Python:
pass
$offEmbeddedCode
$offEcho
$call.checkErrorLevel gams ec11 lo=2 cdir=1234567890123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890


*Test 12: Test for specifying source encoding

* UTF-8 is the default
$onEmbeddedCode Python:
with open('ec_utf8.inc', 'w', encoding='utf-8') as f:
    f.write("$onEmbeddedCode Python:\n")
    f.write("s = 'öäü'\n")
    f.write("$offEmbeddedCode")
$offEmbeddedCode
$include ec_utf8.inc

* Change encoding to cp1252
$onEmbeddedCode Python:
with open('ec_cp1252.inc', 'w', encoding='cp1252') as f:
    f.write("$onEmbeddedCode Python:\n")
    f.write("# coding=cp1252\n")
    f.write("s = 'öäü'\n")
    f.write("$offEmbeddedCode")
$offEmbeddedCode
$include ec_cp1252.inc

* cp1252 is still set from the previous embedded code section
$onEmbeddedCode Python:
with open('ec_cp1252_2.inc', 'w', encoding='cp1252') as f:
    f.write("$onEmbeddedCode Python:\n")
    f.write("s = 'öäü'\n")
    f.write("$offEmbeddedCode")
$offEmbeddedCode
$include ec_cp1252_2.inc

* need to revert to UTF-8
$onEmbeddedCode Python:
with open('ec_utf8_2.inc', 'w', encoding='utf-8') as f:
    f.write("$onEmbeddedCode Python:\n")
    f.write("# coding=utf-8\n")
    f.write("s = 'öäü'\n")
    f.write("$offEmbeddedCode")
$offEmbeddedCode
$include ec_utf8_2.inc

*Test 13: Test for correct alias to universe
$onEcho > ec13.gms
Set s(*); Alias (*,uni);
$echo Alias (uni, *); > unialias_expect.txt
$onEmbeddedCode Python:
gams.wsWorkingDir = '.'
gams.db.export('unialias.gdx')
$offEmbeddedCode
$call.checkErrorLevel gdxdump unialias.gdx | grep Alias > unialias.txt
$call.checkErrorLevel diff unialias.txt unialias_expect.txt > %system.NullFile%
$offEcho
$call.checkErrorLevel gams ec13.gms lo=2

*Test 14: Test gams.printLog with lo=0
$onEcho > test_printlog.gms
$onEmbeddedCode Python:
gams.printLog("some log")
$offEmbeddedCode
$offEcho

$call.checkErrorLevel gams test_printlog.gms lo=0

*Test 15: Test printLog with "%"
$onEchoV > print_log_percent.gms
$onEmbeddedCode Python:
gams.printLog("%file% was not specified")
$offEmbeddedCode
$offEcho
$call.checkErrorLevel gams print_log_percent.gms > print_log_percent.log
$onEmbeddedCode Python:
with open("print_log_percent.log", "r")as f:
  if f.read().count("%file%") != 1:
    raise Exception('Expected "%file%" to be contained in log output once')
$offEmbeddedCode

*Test 16: Test gams.printLog end parameter
$onEcho > test_printlog_newline.gms
$onEmbeddedCode Python:
gams.printLog("s1")
gams.printLog("s2")
gams.printLog("s3", end="\n")
gams.printLog("s4", end="")
gams.printLog("s5", end="x")
$offEmbeddedCode
$offEcho

$call.checkErrorLevel gams test_printlog_newline.gms > log_newline.txt

$onEmbeddedCode Python:
with open("log_newline.txt", "r") as f:
    s = f.read()
if "s1\ns2\ns3\ns4s5x" not in s:
    raise Exception("Unexpected log content.")
$offEmbeddedCode

*Test 17: Test gams.transfer with alias domain
$onEcho > gt14.gms
set i /1/; alias (i,j);
$gdxUnload xxx.gdx

$onEmbeddedCode Python:
import gams.transfer as gt
from gams import GamsWorkspace
ws = GamsWorkspace(working_directory='.', system_directory=r'%gams.sysDir% '.strip())
db = ws.add_database_from_gdx('xxx.gdx')
m = gt.Container(db, system_directory=r'%gams.sysDir% '.strip())
$offEmbeddedCode
$offEcho

$call.checkErrorLevel gams gt14.gms lo=0

*Test 18: Test EC arguments containing quotes
$onEmbeddedCode Python: argument containing 'single' and "double" quotes
args = gams.arguments
if args != '''argument containing 'single' and "double" quotes''':
  raise Exception("gams.arguments not as expected")
$offEmbeddedCode