1 files changed, 17 insertions, 0 deletions
diff --git a/test/kazius-alerts.rb b/test/kazius-alerts.rb
new file mode 100644
index 0000000..ac2f098
--- /dev/null
+++ b/test/kazius-alerts.rb
@@ -0,0 +1,17 @@
+require 'minitest/autorun'
+#require 'kazius-alerts'
+require_relative '../lib/kazius-alerts.rb'
+
+class KaziusAlertsTest < MiniTest::Test
+
+  def test_alerts_size
+    assert_equal 29, KaziusAlerts::SMARTS.size
+  end
+
+  def test_kazius_alerts_prediction
+    prediction = KaziusAlerts.predict("c1ccccc1NN")
+    assert prediction[:prediction]
+    assert_equal [["unsubstituted heteroatom-bonded heteroatom", "[OH,NH2][N,O]", "O=N(O)[O-]"]], prediction[:matches]
+  end
+
+end