IF YOU WOULD LIKE TO GET AN ACCOUNT, please write an email to s dot adaszewski at gmail dot com. User accounts are meant only to report issues and/or generate pull requests. This is a purpose-specific Git hosting for ADARED projects. Thank you for your understanding!
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

50 lines
1.6KB

  1. from icosagon import Data
  2. import torch
  3. import pytest
  4. def test_data_01():
  5. d = Data()
  6. d.add_node_type('Gene', 1000)
  7. d.add_node_type('Drug', 100)
  8. dummy_0 = torch.zeros((100, 1000))
  9. dummy_1 = torch.zeros((1000, 100))
  10. dummy_2 = torch.zeros((100, 100))
  11. dummy_3 = torch.zeros((1000, 1000))
  12. d.add_relation_type('Target', 1, 0, dummy_0)
  13. d.add_relation_type('Interaction', 0, 0, dummy_3)
  14. d.add_relation_type('Side Effect: Nausea', 1, 1, dummy_2)
  15. d.add_relation_type('Side Effect: Infertility', 1, 1, dummy_2)
  16. d.add_relation_type('Side Effect: Death', 1, 1, dummy_2)
  17. print(d)
  18. def test_data_02():
  19. d = Data()
  20. d.add_node_type('Gene', 1000)
  21. d.add_node_type('Drug', 100)
  22. dummy_0 = torch.zeros((100, 1000))
  23. dummy_1 = torch.zeros((1000, 100))
  24. dummy_2 = torch.zeros((100, 100))
  25. dummy_3 = torch.zeros((1000, 1000))
  26. with pytest.raises(ValueError):
  27. d.add_relation_type('Target', 1, 0, dummy_1)
  28. with pytest.raises(ValueError):
  29. d.add_relation_type('Interaction', 0, 0, dummy_2)
  30. with pytest.raises(ValueError):
  31. d.add_relation_type('Side Effect: Nausea', 1, 1, dummy_3)
  32. with pytest.raises(ValueError):
  33. d.add_relation_type('Side Effect: Infertility', 1, 1, dummy_3)
  34. with pytest.raises(ValueError):
  35. d.add_relation_type('Side Effect: Death', 1, 1, dummy_3)
  36. print(d)
  37. def test_data_03():
  38. d = Data()
  39. d.add_node_type('Gene', 1000)
  40. d.add_node_type('Drug', 100)
  41. with pytest.raises(ValueError):
  42. d.add_relation_type('Target', 1, 0, None)
  43. print(d)