Fix tests

coder543 · coder543 · commit fea61d38fe97 · 2017-10-13T19:13:28.000-04:00
diff --git a/src/lib.rs b/src/lib.rs
@@ -7,6 +7,7 @@
 //! use dataplotlib::util::{linspace, zip2};
 //! use dataplotlib::plotbuilder::PlotBuilder2D;
 //! use dataplotlib::plotter::Plotter;
+//! use dataplotlib::draw_sdl::DrawSDL;
 //!
 //! fn main() {
 //!     let x = linspace(0, 10, 100);
@@ -21,11 +22,14 @@
 //!
 //!     // Adds the sin plot and the linear plot with custom colors
 //!     pb.add_color_xy(xy_sin, [1.0, 0.0, 0.0, 1.0]);
-//!     pb.add_color_xy(xy_lin, [0.0, 0.0, 1.0, 1.0]);
+//!     pb.add_color_xy(xy_lin, [0.0, 0.75, 0.0, 1.0]);
+//!
+//!     let sdlh = dataplotlib::sdl2_init();
+//!     let sdl2_window = DrawSDL::new(sdlh);
 //!
 //!     let mut plt = Plotter::new();
-//!     plt.plot2d(pb);
-//!     plt.join();
+//!     plt.plot2d(pb, sdl2_window);
+//! # plt.disown(); // make sure the doc test doesn't last forever
 //! }
 //! ```
 
diff --git a/src/plotter.rs b/src/plotter.rs
@@ -59,6 +59,8 @@ mod test {
     use super::*;
     use plotbuilder::*;
     use util::*;
+    use draw_sdl::DrawSDL;
+    use sdl2_mt;
 
     #[test]
     fn plot2d_test() {
@@ -67,10 +69,13 @@ mod test {
         let y = (&x).iter().map(|x| x.sin()).collect();
         let xy = zip2(&x, &y);
 
+        let sdlh = sdl2_mt::init();
+        let sdl2_window = DrawSDL::new(sdlh);
+
         let mut pb1 = PlotBuilder2D::new();
         pb1.add_simple_xy(xy);
         let mut plt = Plotter::new();
-        plt.plot2d(pb1);
-        plt.join();
+        plt.plot2d(pb1, sdl2_window);
+        plt.disown();
     }
 }
