| 12 | |
| 13 | int get_file_size ( const char *fname ) { |
| 14 | struct stat st; |
| 15 | if ( stat ( fname, &st ) ) { |
| 16 | std::string msg = "can't stat "; |
| 17 | msg += fname; |
| 18 | throw std::runtime_error ( msg ); |
| 19 | } |
| 20 | |
| 21 | int sz = static_cast<int>(st.st_size); |
| 22 | if ( sz != st.st_size ) { |
| 23 | std::string msg = fname; |
| 24 | msg += " too big"; |
| 25 | throw std::runtime_error ( msg ); |
| 26 | } |
| 27 | |
| 28 | return sz; |
| 29 | } |
| 30 | |
| 31 | TEST ( test_memlog ) { |
| 32 | RV th = RV ( "{alog blog }" ); |
| 33 | RV r = RV ( "{r }", "2" ); |
| 34 | |
| 35 | string ls ( "test_log.bin" ); |
| 36 | remove_all ( ls.c_str() ); |
| 37 | |
| 38 | memlog logger ( 100, ls ); |
| 39 | |
| 40 | int rid = logger.add ( r, "" ); |
| 41 | int thid = logger.add ( th, "th" ); |
| 42 | |
| 43 | logger.init(); |
| 44 | |
| 45 | for ( int i = 0; i < 10; i++ ) { |
| 46 | logger.logit ( rid, vec_2 ( ( double ) i, ( double ) ( i + 1 ) ) ); |
| 47 | logger.logit ( thid, vec_2 ( ( double ) ( 100 - i ), ( double ) ( i - 50 ) ) ); |
| 48 | logger.step(); |
| 49 | } |
| 50 | |
| 51 | logger.finalize(); |
| 52 | |
| 53 | CHECK_EQUAL ( get_file_size ( "test_log.matrix" ), get_file_size ( ls.c_str() ) ); |
| 54 | } |