blob: 6b09c4a6930dbe7cfb2e93a30fec02de97d3a0e1 (
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
34
35
36
37
38
39
40
41
42
43
44
45
46
|
require 'test_helper'
class OpentoxRubyApiWrapperTest < Test::Unit::TestCase
def setup
if ENV['LOCAL']
port = 5000
[ "opentox-compound", "opentox-feature" , "opentox-dataset" , "opentox-fminer" , "opentox-lazar" ].each do |component|
ENV[component.upcase.gsub(/-/,'_')] = "http://localhost:#{port}/"
port += 1
end
end
end
def test_create_dataset_and_model_and_make_a_prediction
dataset = OpenTox::Dataset.new :name => "Hamster Carcinogenicity", :filename => "test/hamster_carcinogenicity.csv"
puts dataset.uri
wait_for_completion dataset
assert_match(/#{ENV['OPENTOX_DATASET']}\d+$/,dataset.uri)
assert_equal("Hamster Carcinogenicity",dataset.name)
assert_equal(true,dataset.finished?)
lazar = OpenTox::Lazar.new :dataset_uri => dataset.uri
puts lazar.uri
wait_for_completion lazar
assert_equal(true,lazar.finished?)
assert_match(/#{ENV['OPENTOX_LAZAR']}model\/\d+$/,lazar.uri)
query_structure = OpenTox::Compound.new :smiles => 'c1ccccc1NN'
puts query_structure.uri
prediction = lazar.predict query_structure
puts prediction.uri
wait_for_completion prediction
puts prediction.classification
puts prediction.confidence
puts prediction.neighbors
puts prediction.features
assert_equal(true, prediction.classification)
assert_match(/0\.\d+/, prediction.confidence.to_s)
end
end
def wait_for_completion(object)
while (!object.finished?)
sleep 1
end
end
|
