summaryrefslogtreecommitdiff
path: root/test/descriptor-long.rb
blob: 7a4c00fdd65d9f792176182ec25096c9eceace0a (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
require_relative "setup.rb"
class DescriptorLongTest < MiniTest::Test

  def test_dataset_all
    # TODO: improve CDK descriptor calculation speed or add timeout
    skip "CDK descriptor calculation takes too long for some compounds"
    dataset = OpenTox::Dataset.from_csv_file File.join(DATA_DIR,"hamster_carcinogenicity.mini.csv")
    d = OpenTox::Algorithm::Descriptor.physchem dataset
    assert_equal dataset.compounds, d.compounds
    assert_equal 332, d.features.size
    assert_equal 332, d.data_entries.first.size
    d.delete
  end

  def test_dataset_openbabel
    # TODO: improve CDK descriptor calculation speed or add timeout
    dataset = Dataset.from_csv_file File.join(DATA_DIR,"hamster_carcinogenicity.mini.csv")
    d = Algorithm::Descriptor.physchem dataset, Algorithm::Descriptor::OBDESCRIPTORS.keys
    assert_equal dataset.compounds, d.compounds
    size = Algorithm::Descriptor::OBDESCRIPTORS.keys.size
    assert_equal size, d.features.size
    assert_equal size, d.data_entries.first.size
    d.delete
  end

end