diff --git a/google-tests/test3.cpp b/google-tests/test3.cpp index 78cfd7c..487b882 100644 --- a/google-tests/test3.cpp +++ b/google-tests/test3.cpp @@ -81,19 +81,26 @@ TEST(SignalTest, ranges) { class DebugRunningQuality : public RunningQuality { protected: virtual void dispatchLocked() { locked = true; } - virtual void dispatchBeat(int idx, bool good, double posCorr) { goods.push_back(good); corrs.push_back(posCorr); } + virtual void dispatchBeat(int idx, bool good, double posCorr) { + if (locked && lockedAt == -1) + lockedAt = idx; + goods.push_back(good); + corrs.push_back(posCorr); + } + int lockedAt; bool locked; std::vector corrs; std::vector goods; public: - DebugRunningQuality(): locked(false) {} + DebugRunningQuality(): lockedAt(-1), locked(false) {} explicit DebugRunningQuality(bool disableSsf): RunningQuality(disableSsf), locked(false) {} virtual ~DebugRunningQuality() {} bool isLocked() { return locked; } std::vector getCorrs() { return corrs; } std::vector getGoods() { return goods; } + int getLockedAt() { return lockedAt; } std::vector getBeatTemplate() { return this->beatTemplate; } }; @@ -210,6 +217,9 @@ TEST(SignalTest, RunningQuality_t2) { EXPECT_TRUE(sqi.isLocked()); EXPECT_TRUE(sqi.getCorrs().size() > 50); + EXPECT_TRUE(sqi.getLockedAt() < 10); + std::cout << "lockedAt=" << sqi.getLockedAt() << std::endl; + std::vector corrs(sqi.getCorrs()); npy_save("test3/ssf_t3_sqi_corrs.npy", corrs);