From 72356545aab6ec7589973966a947d28837716d71 Mon Sep 17 00:00:00 2001
From: Stanislaw Adaszewski <stanislaw.adaszewski@roche.com>
Date: Tue, 16 Jun 2020 16:57:22 +0200
Subject: [PATCH] Remove loss as no longer necessary.

---
 src/icosagon/{ => unused}/loss.py        | 0
 tests/icosagon/test_batch.py             | 8 +++++++-
 tests/icosagon/{ => unused}/test_loss.py | 0
 3 files changed, 7 insertions(+), 1 deletion(-)
 rename src/icosagon/{ => unused}/loss.py (100%)
 rename tests/icosagon/{ => unused}/test_loss.py (100%)

diff --git a/src/icosagon/loss.py b/src/icosagon/unused/loss.py
similarity index 100%
rename from src/icosagon/loss.py
rename to src/icosagon/unused/loss.py
diff --git a/tests/icosagon/test_batch.py b/tests/icosagon/test_batch.py
index fa6a4a9..3d185e4 100644
--- a/tests/icosagon/test_batch.py
+++ b/tests/icosagon/test_batch.py
@@ -31,7 +31,7 @@ def test_predictions_batch_01():
 
     rel_pred = RelationPredictions(
         TrainValTest(torch.tensor([1, 0, 1, 0, 1], dtype=torch.float32), torch.zeros(0), torch.zeros(0)),
-        TrainValTest(torch.zeros(0), torch.zeros(0), torch.zeros(0)),
+        TrainValTest(torch.tensor([1, 0, 1, 0, 1], dtype=torch.float32), torch.zeros(0), torch.zeros(0)),
         TrainValTest(torch.zeros(0), torch.zeros(0), torch.zeros(0)),
         TrainValTest(torch.zeros(0), torch.zeros(0), torch.zeros(0))
     )
@@ -40,7 +40,13 @@ def test_predictions_batch_01():
 
     batch = PredictionsBatch(pred, part_type='train', batch_size=1)
     count = 0
+    lst = []
     for (input, target) in batch:
+        assert len(input) == 1
+        assert len(target) == 1
+        lst.append((input[0], target[0]))
         count += 1
+    assert lst == [ (1, 1), (0, 1), (1, 1), (0, 1), (1, 1),
+        (1, 0), (0, 0), (1, 0), (0, 0), (1, 0) ]
 
     assert count == 10
diff --git a/tests/icosagon/test_loss.py b/tests/icosagon/unused/test_loss.py
similarity index 100%
rename from tests/icosagon/test_loss.py
rename to tests/icosagon/unused/test_loss.py