File cleanup for unit tests

This commit is contained in:
vandomej 2021-06-24 06:28:26 -07:00
parent 0ae5cc81e0
commit 4ea2d4cbbc
2 changed files with 9 additions and 1 deletions

View file

@ -60,6 +60,8 @@ fn test_new() {
format!("{{\"tree\":{},\"step\":0,\"iteration_scaling\":{{\"enumType\":\"Linear\",\"enumContent\":1}}}}", format!("{{\"tree\":{},\"step\":0,\"iteration_scaling\":{{\"enumType\":\"Linear\",\"enumContent\":1}}}}",
btree!(TestState::new(0.0))) btree!(TestState::new(0.0)))
); );
std::fs::remove_file("./temp").expect("Unable to remove file");
} }
#[test] #[test]
@ -90,4 +92,6 @@ fn test_run() {
Some(btree!(TestState::new(6.0))) Some(btree!(TestState::new(6.0)))
)) ))
); );
std::fs::remove_file("./temp").expect("Unable to remove file");
} }

View file

@ -1,9 +1,11 @@
use super::super::file_linked::FileLinked; use super::super::file_linked::FileLinked;
use std::fs;
#[test] #[test]
fn test_mutate() -> Result<(), String> { fn test_mutate() -> Result<(), String> {
let tree = btree!(1, Some(btree!(2)), Some(btree!(3, Some(btree!(4)),))); let tree = btree!(1, Some(btree!(2)), Some(btree!(3, Some(btree!(4)),)));
let mut linked_tree = FileLinked::new(tree, String::from("blah.txt"))?; let mut linked_tree = FileLinked::new(tree, String::from("test.txt"))?;
assert_eq!( assert_eq!(
format!("{}", linked_tree.readonly()), format!("{}", linked_tree.readonly()),
@ -28,5 +30,7 @@ fn test_mutate() -> Result<(), String> {
"{\"val\":10,\"left\":{\"val\":13,\"left\":null,\"right\":null},\"right\":{\"val\":3,\"left\":{\"val\":4,\"left\":null,\"right\":null},\"right\":null}}" "{\"val\":10,\"left\":{\"val\":13,\"left\":null,\"right\":null},\"right\":{\"val\":3,\"left\":{\"val\":4,\"left\":null,\"right\":null},\"right\":null}}"
); );
fs::remove_file("test.txt").expect("Unable to remove file");
Ok(()) Ok(())
} }