#! /usr/bin/ruby
# -*- coding: utf-8 -*-

require 'minitest/spec'
require 'minitest/autorun'
require_relative '../lib/y_petri'     # tested component itself
# require 'y_petri'
# require 'sy'

include Pyper if require 'pyper'

describe YPetri::Workspace do
  before do
    @w = YPetri::Workspace.new
    a = @w.Place.new!( default_marking: 1.0, name: "AA" )
    b = @w.Place.new!( default_marking: 2.0, name: "BB" )
    c = @w.Place.new!( ɴ: "CC", default_marking: 3.0 )
    t1 = @w.Transition.new! s: { a => -1, b => -1, c => 1 },
                            rate: 0.1,
                            ɴ: "AA_BB_assembly"
    t2 = @w.Transition.new! ɴ: "AA_appearing",
                            codomain: a,
                            rate: -> { 0.1 },
                            stoichiometry: 1
    @pp, @tt = [a, b, c], [t1, t2]
    @f_name = "test_output.csv"
    @w.set_imc @pp.τBᴍHτ( &:default_marking )
    @w.set_ssc step: 0.1, sampling: 10, time: 0..50
    @w.set_cc( {} )
    @sim = @w.new_timed_simulation
    File.delete @f_name rescue nil
  end

  it "should present places, transitions, nets, simulations" do
    assert_kind_of YPetri::Net, @w.Net::Top
    assert_equal @pp[0], @w.place( "AA" )
    assert_equal :AA, @w.pl( @pp[0] )
    assert_equal @tt[0], @w.transition( "AA_BB_assembly" )
    assert_equal :AA_appearing, @w.tr( @tt[1] )
    assert_equal @pp, @w.places
    assert_equal @tt, @w.transitions
    assert_equal 1, @w.nets.size
    assert_equal 1, @w.simulations.size
    assert_equal 0, @w.cc.size
    assert_equal 3, @w.imc.size
    assert [0.1, 10, 50].each { |e| @w.ssc.include? e }
    assert_equal @sim, @w.simulation
    assert_equal [:Base], @w.clamp_collections.keys
    assert_equal [:Base], @w.initial_marking_collections.keys
    assert_equal [:Base], @w.simulation_settings_collections.keys
    assert_equal [:AA, :BB, :CC], @w.pp
    assert_equal [:AA_BB_assembly, :AA_appearing], @w.tt
    assert_equal [:Top], @w.nn
  end

  it "should simulate" do
    assert_equal 1, @w.simulations.size
    assert_kind_of( YPetri::Simulation, @w.simulation )
    assert_equal 2, @w.simulation.SR_transitions.size
    @tt[0].domain.must_equal [ @pp[0], @pp[1] ]
    @tt[1].domain.must_equal []
    assert_equal [0.2, 0.1], @w.simulation.φ.column_to_a
    @w.simulation.step!
    @w.simulation.run!
    rec_string = @w.simulation.recording_csv_string
    expected_recording_string =
      "0.0,1.0,2.0,3.0\n" +
      "10.0,0.86102,0.86102,4.13898\n" +
      "20.0,1.29984,0.29984,4.70016\n"
    assert rec_string.start_with?( expected_recording_string )
  end
end