From 659c4440918e2e1c68cf388e9d9c7eb3e83eb4d7 Mon Sep 17 00:00:00 2001
From: Nathanael Sensfelder <SpamShield0@MultiAgentSystems.org>
Date: Sat, 7 Nov 2020 14:14:44 +0100
Subject: Adds more unit tests.

---
 .../tonkadur/fate/v1/lang/computation/Cast.java    | 80 +++++++++-------------
 1 file changed, 34 insertions(+), 46 deletions(-)

(limited to 'src')

diff --git a/src/core/src/tonkadur/fate/v1/lang/computation/Cast.java b/src/core/src/tonkadur/fate/v1/lang/computation/Cast.java
index 3906499..8e283c9 100644
--- a/src/core/src/tonkadur/fate/v1/lang/computation/Cast.java
+++ b/src/core/src/tonkadur/fate/v1/lang/computation/Cast.java
@@ -3,6 +3,7 @@ package tonkadur.fate.v1.lang.computation;
 import java.util.Collection;
 import java.util.Collections;
 import java.util.HashMap;
+import java.util.HashSet;
 import java.util.Map;
 import java.util.Set;
 
@@ -24,55 +25,41 @@ public class Cast extends Computation
 
    static
    {
+      Set allowed_targets;
+
       allowed_type_changes = new HashMap<Type,Set<Type>>();
 
-      allowed_type_changes.put
-      (
-         Type.INT,
-         Type.NUMBER_TYPES
-      );
+      /** INT to ... **********************************************************/
+      allowed_targets = new HashSet<Type>();
+      allowed_targets.add(Type.FLOAT);
+      allowed_targets.add(Type.INT);
+      allowed_targets.add(Type.STRING);
 
-      allowed_type_changes.put
-      (
-         Type.FLOAT,
-         Type.NUMBER_TYPES
-      );
+      allowed_type_changes.put(Type.INT, allowed_targets);
 
-      allowed_type_changes.put
-      (
-         Type.DICT,
-         Collections.emptySet()
-      );
+      /** FLOAT to ... ********************************************************/
+      allowed_targets = new HashSet<Type>();
+      allowed_targets.add(Type.FLOAT);
+      allowed_targets.add(Type.INT);
+      allowed_targets.add(Type.STRING);
 
-      allowed_type_changes.put
-      (
-         Type.SET,
-         Collections.emptySet()
-      );
+      allowed_type_changes.put(Type.FLOAT, allowed_targets);
 
-      allowed_type_changes.put
-      (
-         Type.LIST,
-         Collections.emptySet()
-      );
+      /** BOOL to ... *********************************************************/
+      allowed_targets = new HashSet<Type>();
+      allowed_targets.add(Type.BOOL);
+      allowed_targets.add(Type.STRING);
 
-      allowed_type_changes.put
-      (
-         Type.BOOL,
-         Collections.emptySet()
-      );
+      allowed_type_changes.put(Type.BOOL, allowed_targets);
 
-      allowed_type_changes.put
-      (
-         Type.ANY,
-         Collections.singleton(Type.ANY)
-      );
+      /** BOOL to ... *********************************************************/
+      allowed_targets = new HashSet<Type>();
+      allowed_targets.add(Type.BOOL);
+      allowed_targets.add(Type.INT);
+      allowed_targets.add(Type.FLOAT);
+      allowed_targets.add(Type.STRING);
 
-      allowed_type_changes.put
-      (
-         Type.STRING,
-         Type.COMPARABLE_TYPES
-      );
+      allowed_type_changes.put(Type.STRING, allowed_targets);
    }
 
    public static Collection<Type> get_allowed_casts_to (final Type t)
@@ -125,14 +112,15 @@ public class Cast extends Computation
          (value.get_type().can_be_used_as(to))
          ||
          (
-            (to.is_base_type())
-            &&
+            allowed_type_changes.containsKey
             (
-               allowed_type_changes.get(to).contains
-               (
-                  value.get_type().get_act_as_type()
-               )
+               value.get_type().get_act_as_type().get_base_type()
             )
+            &&
+            allowed_type_changes.get
+            (
+               value.get_type().get_act_as_type().get_base_type()
+            ).contains(to.get_act_as_type())
          )
       )
       {
-- 
cgit v1.2.3-70-g09d2