Add ColorSpace cts tests for getDataSpace and createFromDataSpace
function

Bug: 220016464
Test: this
Change-Id: I49f1bede90e93f0cbd42f860cfbd6b0a40e4862b
diff --git a/tests/tests/graphics/src/android/graphics/cts/ColorSpaceTest.java b/tests/tests/graphics/src/android/graphics/cts/ColorSpaceTest.java
index 3d7561c..d52643f 100644
--- a/tests/tests/graphics/src/android/graphics/cts/ColorSpaceTest.java
+++ b/tests/tests/graphics/src/android/graphics/cts/ColorSpaceTest.java
@@ -24,6 +24,7 @@
 import static org.junit.Assert.fail;
 
 import android.graphics.ColorSpace;
+import android.hardware.DataSpace;
 
 import androidx.test.filters.SmallTest;
 
@@ -938,6 +939,25 @@
         }
     }
 
+    @Test
+    public void getDataSpaceFromColorSpace() {
+        ColorSpace cs = ColorSpace.get(ColorSpace.Named.BT709);
+        assertNotNull(cs);
+        assertEquals(DataSpace.DATASPACE_BT709, cs.getDataSpace());
+
+        cs = ColorSpace.get(ColorSpace.Named.ACES);
+        assertEquals(DataSpace.DATASPACE_UNKNOWN, cs.getDataSpace());
+    }
+
+    @Test
+    public void getColorSpaceFromDataSpace() {
+        ColorSpace cs = ColorSpace.getFromDataSpace(DataSpace.DATASPACE_SRGB);
+        assertNotNull(cs);
+        assertEquals(DataSpace.DATASPACE_SRGB, cs.getDataSpace());
+
+        assertNull(ColorSpace.getFromDataSpace(DataSpace.DATASPACE_JFIF));
+    }
+
     @SuppressWarnings("SameParameterValue")
     private void assertArrayNotEquals(float[] a, float[] b, float eps) {
         for (int i = 0; i < a.length; i++) {