Add tests for ascii to float and reverse conversion

jloveric · Dec 2, 2023 · d02704b · d02704b
1 parent 0abd22e
commit d02704b
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/tests/test_datasets.py b/tests/test_datasets.py
@@ -4,11 +4,21 @@
     dataset_from_file,
     SingleTextDataset,
     RandomizeCharacters,
-    TextTransformerDataset
+    TextTransformerDataset,
+    ascii_to_float,
+    float_to_ascii
 )
 from torch.utils.data import DataLoader, Dataset
 import torch
 
+def test_ascii_to_float():
+    sample = torch.arange(128)
+    res = ascii_to_float(sample)
+    assert torch.all(torch.max(res)<1)
+    assert torch.all(torch.min(res)>-1)
+
+    reverse = float_to_ascii(res)
+    assert torch.all(sample==reverse)
 
 def test_dataset_from_gutenberg():
     num_features = 10