summaryrefslogtreecommitdiff
path: root/scripts/create-test.rb
blob: 151b5e80e5e5f785bec215f32bb7c2c50e5a1535 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
#!/usr/bin/env ruby
require_relative '../../lazar/lib/lazar'
include OpenTox

old = Dataset.from_csv_file File.join("data","mazzatorta_log10.csv")
new = Dataset.from_csv_file File.join("data","swiss_log10.csv")

common_compounds = (old.compounds & new.compounds).uniq

puts ["SMILES","-log10(LOAEL)","Dataset"].join ","
data = []
common_compounds.each do |c|
  old_values = old.values(c,old.features.first)
  new_values = new.values(c,new.features.first)
  identical = old_values & new_values
  unless identical.empty?
    old_values -= identical
    new_values -= identical
  end
  identical.each do |v|
    data << [c.smiles,v,"mazzatorta and swiss"] if v
  end
  old_values.each do |v|
    data << [c.smiles,v,"mazzatorta"] if v
  end
  new_values.each do |v|
    data << [c.smiles,v,"swiss"] if v
  end
end

data.sort!{|a,b| a[1] <=> b[1]}

puts data.collect{|r| r.join ","}.join "\n"