vcf/io/writer/record/samples/sample/value: Move array tests to module

zaeleus · Sep 12, 2024 · 2309de3 · 2309de3
1 parent f302a7d
commit 2309de3
Show file tree

Hide file tree

Showing 2 changed files with 48 additions and 25 deletions.
diff --git a/noodles-vcf/src/io/writer/record/samples/sample/value.rs b/noodles-vcf/src/io/writer/record/samples/sample/value.rs
@@ -56,31 +56,6 @@ mod tests {
         let value_buf = ValueBuf::from(vec![Some(8)]);
         t(&mut buf, &header, (&value_buf).into(), b"8")?;
 
-        let value_buf = ValueBuf::from(vec![Some(8), Some(13), None]);
-        t(&mut buf, &header, (&value_buf).into(), b"8,13,.")?;
-
-        let value_buf = ValueBuf::from(vec![Some(0.333)]);
-        t(&mut buf, &header, (&value_buf).into(), b"0.333")?;
-
-        let value_buf = ValueBuf::from(vec![Some(0.333), Some(0.667), None]);
-        t(&mut buf, &header, (&value_buf).into(), b"0.333,0.667,.")?;
-
-        let value_buf = ValueBuf::from(vec![Some('n')]);
-        t(&mut buf, &header, (&value_buf).into(), b"n")?;
-
-        let value_buf = ValueBuf::from(vec![Some('n'), Some('d'), None]);
-        t(&mut buf, &header, (&value_buf).into(), b"n,d,.")?;
-
-        let value_buf = ValueBuf::from(vec![Some(String::from("noodles"))]);
-        t(&mut buf, &header, (&value_buf).into(), b"noodles")?;
-
-        let value_buf = ValueBuf::from(vec![
-            Some(String::from("noodles")),
-            Some(String::from("vcf")),
-            None,
-        ]);
-        t(&mut buf, &header, (&value_buf).into(), b"noodles,vcf,.")?;
-
         Ok(())
     }
 }
diff --git a/noodles-vcf/src/io/writer/record/samples/sample/value/array.rs b/noodles-vcf/src/io/writer/record/samples/sample/value/array.rs
@@ -65,3 +65,51 @@ where
 
     Ok(())
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::variant::record_buf::samples::sample::value::Array as ArrayBuf;
+
+    #[test]
+    fn test_write_array() -> io::Result<()> {
+        fn t(buf: &mut Vec<u8>, array: &ArrayBuf, expected: &[u8]) -> io::Result<()> {
+            buf.clear();
+            write_array(buf, &array.into())?;
+            assert_eq!(buf, expected);
+            Ok(())
+        }
+
+        let mut buf = Vec::new();
+
+        let array = ArrayBuf::Integer(vec![Some(8)]);
+        t(&mut buf, &array, b"8")?;
+
+        let array = ArrayBuf::Integer(vec![Some(8), Some(13), None]);
+        t(&mut buf, &array, b"8,13,.")?;
+
+        let array = ArrayBuf::Float(vec![Some(0.333)]);
+        t(&mut buf, &array, b"0.333")?;
+
+        let array = ArrayBuf::Float(vec![Some(0.333), Some(0.667), None]);
+        t(&mut buf, &array, b"0.333,0.667,.")?;
+
+        let array = ArrayBuf::Character(vec![Some('n')]);
+        t(&mut buf, &array, b"n")?;
+
+        let array = ArrayBuf::Character(vec![Some('n'), Some('d'), None]);
+        t(&mut buf, &array, b"n,d,.")?;
+
+        let array = ArrayBuf::String(vec![Some(String::from("noodles"))]);
+        t(&mut buf, &array, b"noodles")?;
+
+        let array = ArrayBuf::String(vec![
+            Some(String::from("noodles")),
+            Some(String::from("vcf")),
+            None,
+        ]);
+        t(&mut buf, &array, b"noodles,vcf,.")?;
+
+        Ok(())
+    }
+}