1 files changed, 27 insertions, 0 deletions
diff --git a/tests/unit/SpacedRepetitionModel/SRModel_pickCard_test.h b/tests/unit/SpacedRepetitionModel/SRModel_pickCard_test.h
new file mode 100644
index 0000000..aacb59a
--- /dev/null
+++ b/tests/unit/SpacedRepetitionModel/SRModel_pickCard_test.h
@@ -0,0 +1,27 @@
+#include "SRModel_test.h"
+
+#include <vector>
+
+using namespace std;
+
+class SRModel_pickCard_Test: public SRModelTest
+{
+public:
+    static const int NoCard = -1;
+    static const int Now = 0;
+    static const double HourAgo;
+
+protected:
+    static QDateTime timeFromDelta(double daysDelta);
+
+protected:
+    void SetUp();
+    void addStudied(double daysDelta, double interval, int grade = 4,
+            int level = StudyRecord::Repeating);
+    void addUnknown(double daysDelta);
+    void addIncorrect(double daysDelta);
+    void addLearning(double daysDelta);
+    void addNew();
+    void assertCurCard(int expCard);
+    void testPicked(const vector<int>& expCards);
+};